chore: update tests for v1.1.0

Escape-Technologies · May 11, 2022 · 506d43e · 506d43e
1 parent 61c42ab
commit 506d43e
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/tests/unit/test_main.py b/tests/unit/test_main.py
@@ -2,7 +2,6 @@
 
 import argparse
 import importlib.metadata
-import sys
 from multiprocessing import cpu_count
 
 from graphinder import __version__
@@ -60,13 +59,11 @@ def test_validate_arguments() -> None:
 
 def test_main() -> None:
     """main test."""
-    sys.argv = ['graphinder']
-    assert main() is False
+
+    assert main([]) is False
 
 
 def test_full_run() -> None:
     """Test a complete run."""
 
-    sys.argv = ['graphinder', '-d', 'example.com']
-
-    assert main() is True
+    assert main(['-d', 'example.com']) is True