Add tests that catch bug

Author: maushumee

pandas/tests/arrays/numpy_/test_numpy.py

@@ -3,6 +3,8 @@
 the interface tests.
 """
 
+from datetime import datetime
+
 import numpy as np
 import pytest
 
@@ -195,6 +197,68 @@ def test_validate_reduction_keyword_args():
         arr.all(keepdims=True)
 
 
+@pytest.mark.parametrize(
+    "value, expectedError",
+    [
+        (True, False),
+        (9, False),
+        (5.5, True),
+        (1 + 2j, True),
+        ("t", True),
+        (datetime.now(), True),
+    ],
+)
+def test_int_arr_validate_setitem_value(value, expectedError):
+    arr = pd.Series(range(5), dtype="int").array
+    if expectedError:
+        with pytest.raises(TypeError):
+            arr._validate_setitem_value(value)
+    else:
+        arr[0] = value
+        assert arr[0] == value
+
+
+@pytest.mark.parametrize(
+    "value, expectedError",
+    [
+        (True, False),
+        (9, False),
+        (5.5, False),
+        (1 + 2j, True),
+        ("t", True),
+        (datetime.now(), True),
+    ],
+)
+def test_float_arr_validate_setitem_value(value, expectedError):
+    arr = pd.Series(range(5), dtype="float").array
+    if expectedError:
+        with pytest.raises(TypeError):
+            arr._validate_setitem_value(value)
+    else:
+        arr[0] = value
+        assert arr[0] == value
+
+
+@pytest.mark.parametrize(
+    "value, expectedError",
+    [
+        (True, False),
+        (9, False),
+        (5.5, False),
+        ("t", False),
+        (datetime.now(), True),
+    ],
+)
+def test_str_arr_validate_setitem_value(value, expectedError):
+    arr = NumpyExtensionArray(np.array(["foo", "bar", "test"], dtype="str"))
+    if expectedError:
+        with pytest.raises(TypeError):
+            arr._validate_setitem_value(value)
+    else:
+        arr[0] = value
+        assert arr[0] == str(value)
+
+
 def test_np_max_nested_tuples():
     # case where checking in ufunc.nout works while checking for tuples
     #  does not