Add test for GenAiPAException for nonzero exit

triton-inference-server · Mar 1, 2024 · e7e3808 · e7e3808
1 parent 5a3f598
commit e7e3808
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/c++/perf_analyzer/genai-pa/tests/test_cli.py b/src/c++/perf_analyzer/genai-pa/tests/test_cli.py
@@ -28,6 +28,8 @@
 
 import pytest
 from genai_pa import parser
+from genai_pa.exceptions import GenAiPAException
+from genai_pa.main import run
 
 
 class TestCLIArguments:
@@ -91,3 +93,7 @@ def test_arguments_model_not_provided(self):
 
         # Check that the exit was unsuccessful
         assert exc_info.value.code != 0
+
+    def test_execption_on_nonzero_exit(self):
+        with pytest.raises(GenAiPAException) as e:
+            _ = run(["-m", "nonexistent_model"])