add one more test

triton-inference-server · Jul 30, 2024 · d6d8b23 · d6d8b23
1 parent 2e339f4
commit d6d8b23
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/c++/perf_analyzer/genai-perf/tests/test_synthetic_image_generator.py b/src/c++/perf_analyzer/genai-perf/tests/test_synthetic_image_generator.py
@@ -123,3 +123,27 @@ def test_base64_encoding_with_different_formats(image_format):
     img_bytes = BytesIO(img_data)
     image = Image.open(img_bytes)
     assert image.format == image_format.name
+
+
+def test_random_image_format():
+    random.seed(123)
+    img1 = SyntheticImageGenerator.create_synthetic_image(
+        image_width_mean=100,
+        image_width_stddev=100,
+        image_height_mean=100,
+        image_height_stddev=100,
+        image_format=None,
+    )
+
+    random.seed(456)
+    img2 = SyntheticImageGenerator.create_synthetic_image(
+        image_width_mean=100,
+        image_width_stddev=100,
+        image_height_mean=100,
+        image_height_stddev=100,
+        image_format=None,
+    )
+
+    # check prefix
+    assert img1.startswith("data:image/png")
+    assert img2.startswith("data:image/jpeg")