test: add benchmark for cli

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · May 8, 2024 · 6b90a11 · 6b90a11
1 parent 5896553
commit 6b90a11
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -4,6 +4,7 @@
 import subprocess
 import sys
 
+import pytest
 import reacnetgenerator
 
 
@@ -14,3 +15,11 @@ def test_module_script():
         [sys.executable, "-m", "reacnetgenerator", "--version"]
     ).decode("ascii")
     assert output.splitlines()[0] == f"ReacNetGenerator v{expected_version}"
+
+
+@pytest.mark.benchmark
+def test_cli():
+    """Test reacnetgenerator command."""
+    subprocess.check_output(
+        ["reacnetgenerator", "-h"]
+    ).decode("ascii")