floating

Author: cmp0xff

pandas-stubs/_typing.pyi

@@ -504,10 +504,7 @@ PyArrowTimestampDtypeArg: TypeAlias = Literal[
     "timestamp[ns][pyarrow]",
 ]
 TimestampDtypeArg: TypeAlias = (
-    PandasTimestampDtypeArg
-    | PandasAstypeTimestampDtypeArg
-    | NumpyTimestampDtypeArg
-    | PyArrowTimestampDtypeArg
+    PandasTimestampDtypeArg | NumpyTimestampDtypeArg | PyArrowTimestampDtypeArg
 )
 # Builtin str type and its string alias
 BuiltinStrDtypeArg: TypeAlias = type[str] | Literal["str"]

tests/arrays/test_boolean_array.py

@@ -25,5 +25,5 @@ def test_constructor() -> None:
 
 
 @pytest.mark.parametrize("dtype", get_dtype(PandasBooleanDtypeArg))
-def test_constructors_dtype(dtype: PandasBooleanDtypeArg):
+def test_constructor_dtype(dtype: PandasBooleanDtypeArg):
     check(assert_type(pd.array([True], dtype=dtype), BooleanArray), BooleanArray)

tests/arrays/test_numpy_extension_array.py

@@ -8,6 +8,7 @@
     BuiltinDtypeArg,
     NumpyNotTimeDtypeArg,
     check,
+    exception_on_platform,
     get_dtype,
 )
 
@@ -40,14 +41,19 @@ def test_constructor() -> None:
 
 
 @pytest.mark.parametrize("dtype", get_dtype(BuiltinDtypeArg | NumpyNotTimeDtypeArg))
-def test_constructors_dtype(dtype: BuiltinDtypeArg | NumpyNotTimeDtypeArg):
+def test_constructor_dtype(dtype: BuiltinDtypeArg | NumpyNotTimeDtypeArg):
     if dtype == "V" or "void" in str(dtype):
         check(
             assert_type(pd.array([b"1"], dtype=dtype), NumpyExtensionArray),
             NumpyExtensionArray,
         )
     else:
-        check(
-            assert_type(pd.array([1], dtype=dtype), NumpyExtensionArray),
-            NumpyExtensionArray,
-        )
+        exc = exception_on_platform(dtype)
+        if exc:
+            with pytest.raises(exc):
+                pd.array([1], dtype=dtype)
+        else:
+            check(
+                assert_type(pd.array([1], dtype=dtype), NumpyExtensionArray),
+                NumpyExtensionArray,
+            )