Skip to content

Commit 80c9273

Browse files
nv-hwoomc-nv
authored andcommitted
fix test (#637)
1 parent 41a3ac2 commit 80c9273

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

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

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,8 @@ class TestPlotConfigParser:
4141
y_metric: inter_token_latencies
4242
x_label: TTFT (ms)
4343
y_label: ITL (ms)
44+
width: 1000
45+
height: 3000
4446
type: box
4547
paths:
4648
- run1/concurrency32.json
@@ -54,6 +56,8 @@ class TestPlotConfigParser:
5456
y_metric: num_output_tokens
5557
x_label: Input Tokens
5658
y_label: Output Tokens
59+
width: 1234
60+
height: 5678
5761
type: scatter
5862
paths:
5963
- run4/concurrency1.json
@@ -78,6 +82,8 @@ def test_generate_configs(self, monkeypatch) -> None:
7882
assert pc1.title == "TTFT vs ITL"
7983
assert pc1.x_label == "TTFT (ms)"
8084
assert pc1.y_label == "ITL (ms)"
85+
assert pc1.width == 1000
86+
assert pc1.height == 3000
8187
assert pc1.type == PlotType.BOX
8288
assert pc1.output == Path("test_output_1")
8389

@@ -94,6 +100,8 @@ def test_generate_configs(self, monkeypatch) -> None:
94100
assert pc2.title == "Num Input Token vs Num Output Token"
95101
assert pc2.x_label == "Input Tokens"
96102
assert pc2.y_label == "Output Tokens"
103+
assert pc2.width == 1234
104+
assert pc2.height == 5678
97105
assert pc2.type == PlotType.SCATTER
98106
assert pc2.output == Path("test_output_2")
99107

0 commit comments

Comments
 (0)