Skip to content

Commit aac6cff

Browse files
committed
fix(mypy): attempt for python > 310
1 parent 6469196 commit aac6cff

File tree

1 file changed

+25
-24
lines changed

1 file changed

+25
-24
lines changed
Lines changed: 25 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from builtins import str as _str
12
from typing import Any
23

34
import numpy as np
@@ -10,60 +11,60 @@
1011

1112

1213
def test_add_py_scalar() -> None:
13-
"""Test pd.Series[str] + Python native str"""
14+
"""Testpd.Series[str]+ Python native str"""
1415
r0 = "right"
1516

16-
check(assert_type(left + r0, "pd.Series[str]"), pd.Series, str)
17+
check(assert_type(left + r0, "pd.Series[_str]"), pd.Series, _str)
1718

18-
check(assert_type(r0 + left, "pd.Series[str]"), pd.Series, str)
19+
check(assert_type(r0 + left, "pd.Series[_str]"), pd.Series, _str)
1920

20-
check(assert_type(left.add(r0), "pd.Series[str]"), pd.Series, str)
21+
check(assert_type(left.add(r0), "pd.Series[_str]"), pd.Series, _str)
2122

22-
check(assert_type(left.radd(r0), "pd.Series[str]"), pd.Series, str)
23+
check(assert_type(left.radd(r0), "pd.Series[_str]"), pd.Series, _str)
2324

2425

2526
def test_add_py_sequence() -> None:
26-
"""Test pd.Series[str] + Python native sequence"""
27+
"""Testpd.Series[str]+ Python native sequence"""
2728
r0 = ["a", "bc", "def"]
2829
r1 = tuple(r0)
2930

30-
check(assert_type(left + r0, "pd.Series[str]"), pd.Series, str)
31-
check(assert_type(left + r1, "pd.Series[str]"), pd.Series, str)
31+
check(assert_type(left + r0, "pd.Series[_str]"), pd.Series, _str)
32+
check(assert_type(left + r1, "pd.Series[_str]"), pd.Series, _str)
3233

33-
check(assert_type(r0 + left, "pd.Series[str]"), pd.Series, str)
34-
check(assert_type(r1 + left, "pd.Series[str]"), pd.Series, str)
34+
check(assert_type(r0 + left, "pd.Series[_str]"), pd.Series, _str)
35+
check(assert_type(r1 + left, "pd.Series[_str]"), pd.Series, _str)
3536

36-
check(assert_type(left.add(r0), "pd.Series[str]"), pd.Series, str)
37-
check(assert_type(left.add(r1), "pd.Series[str]"), pd.Series, str)
37+
check(assert_type(left.add(r0), "pd.Series[_str]"), pd.Series, _str)
38+
check(assert_type(left.add(r1), "pd.Series[_str]"), pd.Series, _str)
3839

39-
check(assert_type(left.radd(r0), "pd.Series[str]"), pd.Series, str)
40-
check(assert_type(left.radd(r1), "pd.Series[str]"), pd.Series, str)
40+
check(assert_type(left.radd(r0), "pd.Series[_str]"), pd.Series, _str)
41+
check(assert_type(left.radd(r1), "pd.Series[_str]"), pd.Series, _str)
4142

4243

4344
def test_add_numpy_array() -> None:
44-
"""Test pd.Series[str] + numpy array"""
45+
"""Testpd.Series[str]+ numpy array"""
4546
r0 = np.array(["a", "bc", "def"], np.str_)
4647

47-
check(assert_type(left + r0, "pd.Series[str]"), pd.Series, str)
48+
check(assert_type(left + r0, "pd.Series[_str]"), pd.Series, _str)
4849

4950
# `numpy` typing gives `Any` in the static type
5051
# checking, where our `__radd__` cannot override. At runtime, they return
5152
# `Series`s.
52-
check(assert_type(r0 + left, Any), pd.Series, str)
53+
check(assert_type(r0 + left, Any), pd.Series, _str)
5354

54-
check(assert_type(left.add(r0), "pd.Series[str]"), pd.Series, str)
55+
check(assert_type(left.add(r0), "pd.Series[_str]"), pd.Series, _str)
5556

56-
check(assert_type(left.radd(r0), "pd.Series[str]"), pd.Series, str)
57+
check(assert_type(left.radd(r0), "pd.Series[_str]"), pd.Series, _str)
5758

5859

5960
def test_add_pd_series() -> None:
60-
"""Test pd.Series[str] + pandas series"""
61+
"""Testpd.Series[str]+ pandas series"""
6162
r0 = pd.Series(["a", "bc", "def"])
6263

63-
check(assert_type(left + r0, "pd.Series[str]"), pd.Series, str)
64+
check(assert_type(left + r0, "pd.Series[_str]"), pd.Series, _str)
6465

65-
check(assert_type(r0 + left, "pd.Series[str]"), pd.Series, str)
66+
check(assert_type(r0 + left, "pd.Series[_str]"), pd.Series, _str)
6667

67-
check(assert_type(left.add(r0), "pd.Series[str]"), pd.Series, str)
68+
check(assert_type(left.add(r0), "pd.Series[_str]"), pd.Series, _str)
6869

69-
check(assert_type(left.radd(r0), "pd.Series[str]"), pd.Series, str)
70+
check(assert_type(left.radd(r0), "pd.Series[_str]"), pd.Series, _str)

0 commit comments

Comments
 (0)