From d7fcbfc2af47fb6430305a4b07865c0aaa2fbd12 Mon Sep 17 00:00:00 2001
From: "J. Sebastian Paez" <jspaezp@users.noreply.github.com>
Date: Sun, 27 Oct 2024 13:09:39 -0700
Subject: [PATCH] Update tests/test_end_to_end.py

Co-authored-by: Florian Rathgeber <florian.rathgeber@gmail.com>
---
 tests/test_end_to_end.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/tests/test_end_to_end.py b/tests/test_end_to_end.py
index 5e8913c..6da5b3a 100644
--- 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)