Add other int types to integer validator tests

marthacryan · marthacryan · commit c2453d331fb8 · 2024-08-07T15:15:06.000-07:00
diff --git a/packages/python/plotly/_plotly_utils/tests/validators/test_integer_validator.py b/packages/python/plotly/_plotly_utils/tests/validators/test_integer_validator.py
@@ -113,9 +113,43 @@ def test_acceptance_aok_list(val, validator_aok):
 
 
 # Test base64 encoded arrays with array_ok=True
-@pytest.mark.parametrize("val", [b64(np.array([1, 0], dtype="int16")), b64([1, 0])])
-def test_acceptance_aok_base64(val, validator_aok):
-    assert np.array_equal(validator_aok.validate_coerce(val), val)
+INT_BASE64_TEST_CASES = [
+    (
+        b64(np.array([1, 2, 3], dtype="int64")),
+        {"bdata": "AQID", "dtype": "i1"},
+    ),
+    (
+        b64(np.array([1, 2, 3], dtype="int32")),
+        {"bdata": "AQAAAAIAAAADAAAA", "dtype": "i4"},
+    ),
+    (
+        b64(np.array([1, 2, 3], dtype="int16")),
+        {"bdata": "AQACAAMA", "dtype": "i2"},
+    ),
+    (
+        b64(np.array([1, 2, 3], dtype="int8")),
+        {"bdata": "AQID", "dtype": "i1"},
+    ),
+    (
+        b64(np.array([1, 2, 3], dtype="uint64")),
+        {"bdata": "AQID", "dtype": "u1"},
+    ),
+    (
+        b64(np.array([1, 2, 3], dtype="uint32")),
+        {"bdata": "AQAAAAIAAAADAAAA", "dtype": "u4"},
+    ),
+    (
+        b64(np.array([1, 2, 3], dtype="uint16")),
+        {"bdata": "AQACAAMA", "dtype": "u2"},
+    ),
+    (
+        b64(np.array([1, 2, 3], dtype="uint8")),
+        {"bdata": "AQID", "dtype": "u1"},
+    ),
+]
+@pytest.mark.parametrize("val, expected", INT_BASE64_TEST_CASES)
+def test_acceptance_aok_base64(val, expected, validator_aok):
+    assert np.array_equal(validator_aok.validate_coerce(val), expected)
 
 
 # ### Coerce ###
