adding tests for pandas-dev#748

mutricyl · May 14, 2024 · 2bd313c · 2bd313c
1 parent a251e5a
commit 2bd313c
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/tests/test_series.py b/tests/test_series.py
@@ -3214,3 +3214,28 @@ def test_diff_never3() -> None:
     if TYPE_CHECKING_INVALID_USAGE:
         # str -> TypeError: unsupported operand type(s) for -: 'str' and 'str'
         assert_never(pd.Series(["a", "b"]).diff())
+
+
+def test_operator_constistency() -> None:
+    # created for #748
+    s = pd.Series([1, 2, 3])
+    check(
+        assert_type(s * np.timedelta64(1, "s"), "TimedeltaSeries"),
+        pd.Series,
+        pd.Timedelta,
+    )
+    check(
+        assert_type(np.timedelta64(1, "s") * s, "TimedeltaSeries"),
+        pd.Series,
+        pd.Timedelta,
+    )
+    check(
+        assert_type(s.mul(np.timedelta64(1, "s")), "TimedeltaSeries"),
+        pd.Series,
+        pd.Timedelta,
+    )
+    check(
+        assert_type(s.rmul(np.timedelta64(1, "s")), "TimedeltaSeries"),
+        pd.Series,
+        pd.Timedelta,
+    )