adjust test_dataarray_validator.py

Author: archmoj

packages/python/plotly/_plotly_utils/tests/validators/test_dataarray_validator.py

@@ -2,6 +2,7 @@
 from _plotly_utils.basevalidators import DataArrayValidator
 import numpy as np
 import pandas as pd
+from plotly.tests.b64 import b64
 
 # Fixtures
 # --------
@@ -33,12 +34,22 @@ def test_validator_acceptance_simple(val, validator):
 
 @pytest.mark.parametrize(
     "val",
-    [np.array([2, 3, 4]), pd.Series(["a", "b", "c"]), np.array([[1, 2, 3], [4, 5, 6]])],
+    [pd.Series(["a", "b", "c"])],
 )
 def test_validator_acceptance_homogeneous(val, validator):
     coerce_val = validator.validate_coerce(val)
     assert isinstance(coerce_val, np.ndarray)
-    assert np.array_equal(validator.present(coerce_val), val)
+    assert np.array_equal(validator.present(coerce_val), b64(val))
+
+
+@pytest.mark.parametrize(
+    "val",
+    [np.array([2, 3, 4]), np.array([[1, 2, 3], [4, 5, 6]])],
+)
+def test_validator_acceptance_homogeneous(val, validator):
+    coerce_val = validator.validate_coerce(val)
+    assert isinstance(coerce_val, object)
+    assert np.array_equal(validator.present(coerce_val), b64(val))
 
 
 # ### Rejection ###