adjust test_xarray_input.py

Author: archmoj

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

@@ -9,6 +9,8 @@
     ColorValidator,
 )
 
+from plotly.tests.b64 import _b64
+
 
 @pytest.fixture
 def data_array_validator(request):
@@ -99,10 +101,12 @@ def test_data_array_validator(data_array_validator, numeric_xarray):
     res = data_array_validator.validate_coerce(numeric_xarray)
 
     # Check type
-    assert isinstance(res, np.ndarray)
+    assert isinstance(res, object)
+
+    numeric_xarray = _b64(numeric_xarray)
 
     # Check dtype
-    assert res.dtype == numeric_xarray.dtype
+    assert res["dtype"] == numeric_xarray["dtype"]
 
     # Check values
     np.testing.assert_array_equal(res, numeric_xarray)