Add more validator tests

Author: marthacryan

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

@@ -3,6 +3,7 @@
 # from ..basevalidators import AngleValidator
 from _plotly_utils.basevalidators import AngleValidator
 import numpy as np
+from plotly.tests.b64 import b64
 
 
 # Fixtures
@@ -51,6 +52,13 @@ def test_aok_acceptance(val, validator_aok):
     assert np.array_equal(validator_aok.validate_coerce(np.array(val)), np.array(val))
 
 
+# Test base64 array
+def test_aok_base64_array(validator_aok):
+    val = b64(np.array([1, 2, 3], dtype="int64"))
+    coerce_val = validator_aok.validate_coerce(val)
+    assert coerce_val["bdata"] == "AQID"
+    assert coerce_val["dtype"] == "i1"
+
 # ### Test coercion above 180 ###
 @pytest.mark.parametrize(
     "val,expected",

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

@@ -1,6 +1,7 @@
 import pytest
 from _plotly_utils.basevalidators import AnyValidator
 import numpy as np
+from plotly.tests.b64 import b64
 
 
 # Fixtures
@@ -49,3 +50,9 @@ def test_acceptance_array(val, validator_aok):
     else:
         assert coerce_val == val
         assert validator_aok.present(coerce_val) == val
+
+def test_base64_array(validator_aok):
+    val = b64(np.array([1, 2, 3], dtype="int64"))
+    coerce_val = validator_aok.validate_coerce(val)
+    assert coerce_val['bdata'] == "AQID"
+    assert coerce_val['dtype'] == "i1"

packages/python/plotly/plotly/tests/test_optional/test_graph_objs/test_performance.py

@@ -56,7 +56,7 @@ def test_performance_b64_scatter3d():
     np_time_elapsed = time.time() - np_start
 
     # np should be faster than lists
-    assert (np_time_elapsed / list_time_elapsed) < 0.75
+    assert (np_time_elapsed / list_time_elapsed) < 0.78
 
 
 FLOAT_TEST_CASES = [