Add io.BytesIO test

Author: maresb

packages/python/plotly/plotly/tests/test_optional/test_kaleido/test_kaleido.py

@@ -2,6 +2,7 @@
 import plotly.io.kaleido
 import sys
 from contextlib import contextmanager
+from io import BytesIO
 
 try:
     from pathlib import Path
@@ -147,3 +148,13 @@ def test_image_renderer():
         height=renderer.height,
         scale=renderer.scale,
     )
+
+
+def test_bytesio():
+    # Verify that writing to a BytesIO object contains the same data as with to_image().
+    bio = BytesIO()
+    pio.write_image(fig, bio, format="jpg", engine="kaleido", validate=False)
+    bio.seek(0)
+    bio_bytes = bio.read()
+    to_image_bytes = pio.to_image(fig, format="jpg", engine="kaleido", validate=False)
+    assert bio_bytes == to_image_bytes

packages/python/plotly/plotly/tests/test_orca/test_to_image.py

@@ -6,6 +6,7 @@
 import pytest
 import sys
 import pandas as pd
+from io import BytesIO
 
 if sys.version_info >= (3, 3):
     from unittest.mock import MagicMock
@@ -339,3 +340,13 @@ def test_invalid_figure_json():
     )
 
     assert "400: invalid or malformed request syntax" in str(err.value)
+
+
+def test_bytesio():
+    # Verify that writing to a BytesIO object contains the same data as with to_image().
+    bio = BytesIO()
+    pio.write_image(fig1, bio, format="jpg", validate=False)
+    bio.seek(0)
+    bio_bytes = bio.read()
+    to_image_bytes = pio.to_image(fig1, format="jpg", validate=False)
+    assert bio_bytes == to_image_bytes