|
27 | 27 | from pandas._typing import Scalar |
28 | 28 |
|
29 | 29 | from tests import ( |
| 30 | + PD_LTE_23, |
30 | 31 | TYPE_CHECKING_INVALID_USAGE, |
31 | 32 | check, |
32 | 33 | ) |
@@ -428,16 +429,18 @@ def test_frame_setitem_na() -> None: |
428 | 429 | df.at["a", "y"] = None |
429 | 430 | df.iat[0, 0] = None |
430 | 431 |
|
431 | | - df.loc[:, "x"] = [None, pd.NA, pd.NaT] |
432 | | - df.iloc[:, 0] = [None, pd.NA, pd.NaT] |
| 432 | + if PD_LTE_23: |
| 433 | + # TODO: pandas-dev/pandas#63420, this is failing on latest build, should work |
| 434 | + df.loc[:, "x"] = [None, pd.NA, pd.NaT] |
| 435 | + df.iloc[:, 0] = [None, pd.NA, pd.NaT] |
433 | 436 |
|
434 | | - # TODO: mypy bug, remove after python/mypy#20420 has been resolved |
435 | | - df.loc[:, ["x"]] = [[None], [pd.NA], [pd.NaT]] # type: ignore[assignment,index] |
436 | | - df.iloc[:, [0]] = [[None], [pd.NA], [pd.NaT]] # type: ignore[assignment,index] |
| 437 | + # TODO: mypy bug, remove after python/mypy#20420 has been resolved |
| 438 | + df.loc[:, ["x"]] = [[None], [pd.NA], [pd.NaT]] # type: ignore[assignment,index] |
| 439 | + df.iloc[:, [0]] = [[None], [pd.NA], [pd.NaT]] # type: ignore[assignment,index] |
437 | 440 |
|
438 | | - # TODO: mypy bug, remove after python/mypy#20420 has been resolved |
439 | | - df.loc[:, iter(["x"])] = [[None], [pd.NA], [pd.NaT]] # type: ignore[assignment,index] |
440 | | - df.iloc[:, iter([0])] = [[None], [pd.NA], [pd.NaT]] # type: ignore[assignment,index] |
| 441 | + # TODO: mypy bug, remove after python/mypy#20420 has been resolved |
| 442 | + df.loc[:, iter(["x"])] = [[None], [pd.NA], [pd.NaT]] # type: ignore[assignment,index] |
| 443 | + df.iloc[:, iter([0])] = [[None], [pd.NA], [pd.NaT]] # type: ignore[assignment,index] |
441 | 444 |
|
442 | 445 |
|
443 | 446 | def test_loc_set() -> None: |
|
0 commit comments