Parametrize test for help CLI flags (#4)

brettcannon · Aug 20, 2019 · e7ce982 · e7ce982
1 parent 84d006d
commit e7ce982
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/test_main.py b/test_main.py
@@ -40,12 +40,12 @@ def call_py(*args):
  yield call_py
 
 
-def test_help(py):
- for flag in ["--help", "-h"]:
-  call = py(flag)
-  assert not call.returncode
-  assert os.fspath(py.path) in call.stdout
-  assert sys.executable in call.stdout
+@pytest.mark.parametrize("flag", ["--help", "-h"])
+def test_help(py, flag):
+ call = py(flag)
+ assert not call.returncode
+ assert os.fspath(py.path) in call.stdout
+ assert sys.executable in call.stdout
 
 
 def test_list(py):