File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
src/c++/perf_analyzer/genai-perf/tests Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments