Skip to content

Commit 49673de

Browse files
nv-hwoomc-nv
authored andcommitted
Add unit test
1 parent 01f0876 commit 49673de

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

src/c++/perf_analyzer/genai-perf/tests/test_cli.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -573,3 +573,17 @@ def test_compare_not_provided(self, monkeypatch, capsys):
573573
assert excinfo.value.code != 0
574574
captured = capsys.readouterr()
575575
assert expected_output in captured.err
576+
577+
@pytest.mark.parametrize(
578+
"args, expected_model",
579+
[
580+
(["--files", "profile1.json", "profile2.json", "profile3.json"], None),
581+
(["--config", "config.yaml"], None),
582+
],
583+
)
584+
def test_compare_model_arg(self, monkeypatch, args, expected_model):
585+
combined_args = ["genai-perf", "compare"] + args
586+
monkeypatch.setattr("sys.argv", combined_args)
587+
args, _ = parser.parse_args()
588+
589+
assert args.model == expected_model

0 commit comments

Comments
 (0)