11from contextlib import contextmanager
22from typing import Iterator , Tuple
33
4- from exercise_utils .test import GitAutograderTest , GitAutograderTestLoader , assert_output
4+ from exercise_utils .test import (
5+ GitAutograderTest ,
6+ GitAutograderTestLoader ,
7+ assert_output ,
8+ )
59from git_autograder import GitAutograderStatus
610from repo_smith .repo_smith import RepoSmith
711
@@ -23,7 +27,7 @@ def base_setup() -> Iterator[Tuple[GitAutograderTest, RepoSmith]]:
2327 with loader .start (include_remote_repo = True ) as (test , rs , rs_remote ):
2428 remote_path = str (rs_remote .repo .git_dir )
2529 rs .git .remote_add ("origin" , remote_path )
26-
30+
2731 rs .git .commit (allow_empty = True , message = "Initial commit" )
2832
2933 rs .git .checkout ("ABC" , branch = True )
@@ -39,6 +43,7 @@ def base_setup() -> Iterator[Tuple[GitAutograderTest, RepoSmith]]:
3943
4044 yield (test , rs )
4145
46+
4247def test_no_changes ():
4348 with base_setup () as (test , rs ):
4449 output = test .run ()
@@ -49,9 +54,10 @@ def test_no_changes():
4954 BRANCH_MISSING .format (branch = "STU" ),
5055 BRANCH_MISSING .format (branch = "VWX" ),
5156 REMOTE_COMMIT_MISSING .format (branch = "ABC" ),
52- REMOTE_COMMIT_MISSING .format (branch = "DEF" )
53- ])
54-
57+ REMOTE_COMMIT_MISSING .format (branch = "DEF" ),
58+ ],
59+ )
60+
5561
5662def test_branch_not_tracking ():
5763 with base_setup () as (test , rs ):
@@ -66,9 +72,10 @@ def test_branch_not_tracking():
6672 BRANCH_NOT_TRACKING .format (branch = "STU" ),
6773 BRANCH_NOT_TRACKING .format (branch = "VWX" ),
6874 REMOTE_COMMIT_MISSING .format (branch = "ABC" ),
69- REMOTE_COMMIT_MISSING .format (branch = "DEF" )
70- ])
71-
75+ REMOTE_COMMIT_MISSING .format (branch = "DEF" ),
76+ ],
77+ )
78+
7279
7380def test_def_local_commit_missing ():
7481 with base_setup () as (test , rs ):
@@ -87,11 +94,8 @@ def test_def_local_commit_missing():
8794 rs .git .merge ("origin/DEF" )
8895
8996 output = test .run ()
90- assert_output (
91- output ,
92- GitAutograderStatus .UNSUCCESSFUL ,
93- [LOCAL_COMMIT_MISSING ])
94-
97+ assert_output (output , GitAutograderStatus .UNSUCCESSFUL , [LOCAL_COMMIT_MISSING ])
98+
9599
96100def test_successful_changes ():
97101 with base_setup () as (test , rs ):
@@ -110,4 +114,4 @@ def test_successful_changes():
110114 rs .git .merge ("origin/DEF" )
111115
112116 output = test .run ()
113- assert_output (output , GitAutograderStatus .SUCCESSFUL )
117+ assert_output (output , GitAutograderStatus .SUCCESSFUL )
0 commit comments