Update tests/test_end_to_end.py

Co-authored-by: Florian Rathgeber <florian.rathgeber@gmail.com>
kynan · Oct 27, 2024 · d7fcbfc · d7fcbfc
1 parent 0955d2b
commit d7fcbfc
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/tests/test_end_to_end.py b/tests/test_end_to_end.py
@@ -120,10 +120,7 @@ def test_dry_run_stdin(input_file: str, extra_args: List[str], verify: bool):
         output = pc.stdout
 
     assert output == expected
-    if verify:
-        assert pc.returncode == 1
-    else:
-        assert pc.returncode == 0
+    assert pc.returncode == 1 if verify else 0
 
 
 @pytest.mark.parametrize("input_file, extra_args", DRY_RUN_CASES)