|
12 | 12 | import numpy as np |
13 | 13 | from numpy import typing as npt |
14 | 14 | import pandas as pd |
15 | | -from pandas.core.indexes.numeric import IntegerIndex |
| 15 | +from pandas.core.indexes.numeric import NumericIndex |
16 | 16 | import pytz |
17 | 17 | from typing_extensions import assert_type |
18 | 18 |
|
@@ -441,37 +441,37 @@ def test_series_dt_accessors() -> None: |
441 | 441 | def test_datetimeindex_accessors() -> None: |
442 | 442 | # GH 194 |
443 | 443 | x = pd.DatetimeIndex(["2022-08-14", "2022-08-20"]) |
444 | | - check(assert_type(x.month, IntegerIndex), IntegerIndex) |
| 444 | + check(assert_type(x.month, NumericIndex), NumericIndex) |
445 | 445 |
|
446 | 446 | i0 = pd.date_range(start="2022-06-01", periods=10) |
447 | 447 | check(assert_type(i0, pd.DatetimeIndex), pd.DatetimeIndex, pd.Timestamp) |
448 | 448 |
|
449 | 449 | check(assert_type(i0.date, np.ndarray), np.ndarray, dt.date) |
450 | 450 | check(assert_type(i0.time, np.ndarray), np.ndarray, dt.time) |
451 | 451 | check(assert_type(i0.timetz, np.ndarray), np.ndarray, dt.time) |
452 | | - check(assert_type(i0.year, IntegerIndex), IntegerIndex, int) |
453 | | - check(assert_type(i0.month, IntegerIndex), IntegerIndex, int) |
454 | | - check(assert_type(i0.day, IntegerIndex), IntegerIndex, int) |
455 | | - check(assert_type(i0.hour, IntegerIndex), IntegerIndex, int) |
456 | | - check(assert_type(i0.minute, IntegerIndex), IntegerIndex, int) |
457 | | - check(assert_type(i0.second, IntegerIndex), IntegerIndex, int) |
458 | | - check(assert_type(i0.microsecond, IntegerIndex), IntegerIndex, int) |
459 | | - check(assert_type(i0.nanosecond, IntegerIndex), IntegerIndex, int) |
460 | | - check(assert_type(i0.dayofweek, IntegerIndex), IntegerIndex, int) |
461 | | - check(assert_type(i0.day_of_week, IntegerIndex), IntegerIndex, int) |
462 | | - check(assert_type(i0.weekday, IntegerIndex), IntegerIndex, int) |
463 | | - check(assert_type(i0.dayofyear, IntegerIndex), IntegerIndex, int) |
464 | | - check(assert_type(i0.day_of_year, IntegerIndex), IntegerIndex, int) |
465 | | - check(assert_type(i0.quarter, IntegerIndex), IntegerIndex, int) |
| 452 | + check(assert_type(i0.year, NumericIndex), NumericIndex, int) |
| 453 | + check(assert_type(i0.month, NumericIndex), NumericIndex, int) |
| 454 | + check(assert_type(i0.day, NumericIndex), NumericIndex, int) |
| 455 | + check(assert_type(i0.hour, NumericIndex), NumericIndex, int) |
| 456 | + check(assert_type(i0.minute, NumericIndex), NumericIndex, int) |
| 457 | + check(assert_type(i0.second, NumericIndex), NumericIndex, int) |
| 458 | + check(assert_type(i0.microsecond, NumericIndex), NumericIndex, int) |
| 459 | + check(assert_type(i0.nanosecond, NumericIndex), NumericIndex, int) |
| 460 | + check(assert_type(i0.dayofweek, NumericIndex), NumericIndex, int) |
| 461 | + check(assert_type(i0.day_of_week, NumericIndex), NumericIndex, int) |
| 462 | + check(assert_type(i0.weekday, NumericIndex), NumericIndex, int) |
| 463 | + check(assert_type(i0.dayofyear, NumericIndex), NumericIndex, int) |
| 464 | + check(assert_type(i0.day_of_year, NumericIndex), NumericIndex, int) |
| 465 | + check(assert_type(i0.quarter, NumericIndex), NumericIndex, int) |
466 | 466 | check(assert_type(i0.is_month_start, npt.NDArray[np.bool_]), np.ndarray, np.bool_) |
467 | 467 | check(assert_type(i0.is_month_end, npt.NDArray[np.bool_]), np.ndarray, np.bool_) |
468 | 468 | check(assert_type(i0.is_quarter_start, npt.NDArray[np.bool_]), np.ndarray, np.bool_) |
469 | 469 | check(assert_type(i0.is_quarter_end, npt.NDArray[np.bool_]), np.ndarray, np.bool_) |
470 | 470 | check(assert_type(i0.is_year_start, npt.NDArray[np.bool_]), np.ndarray, np.bool_) |
471 | 471 | check(assert_type(i0.is_year_end, npt.NDArray[np.bool_]), np.ndarray, np.bool_) |
472 | 472 | check(assert_type(i0.is_leap_year, npt.NDArray[np.bool_]), np.ndarray, np.bool_) |
473 | | - check(assert_type(i0.daysinmonth, IntegerIndex), IntegerIndex, int) |
474 | | - check(assert_type(i0.days_in_month, IntegerIndex), IntegerIndex, int) |
| 473 | + check(assert_type(i0.daysinmonth, NumericIndex), NumericIndex, int) |
| 474 | + check(assert_type(i0.days_in_month, NumericIndex), NumericIndex, int) |
475 | 475 | check(assert_type(i0.tz, Optional[dt.tzinfo]), type(None)) |
476 | 476 | check(assert_type(i0.freq, Optional[BaseOffset]), BaseOffset) |
477 | 477 | check(assert_type(i0.isocalendar(), pd.DataFrame), pd.DataFrame) |
@@ -531,20 +531,20 @@ def test_periodindex_accessors() -> None: |
531 | 531 | i0 = pd.period_range(start="2022-06-01", periods=10) |
532 | 532 | check(assert_type(i0, pd.PeriodIndex), pd.PeriodIndex, pd.Period) |
533 | 533 |
|
534 | | - check(assert_type(i0.year, IntegerIndex), IntegerIndex, int) |
535 | | - check(assert_type(i0.month, IntegerIndex), IntegerIndex, int) |
536 | | - check(assert_type(i0.day, IntegerIndex), IntegerIndex, int) |
537 | | - check(assert_type(i0.hour, IntegerIndex), IntegerIndex, int) |
538 | | - check(assert_type(i0.minute, IntegerIndex), IntegerIndex, int) |
539 | | - check(assert_type(i0.second, IntegerIndex), IntegerIndex, int) |
540 | | - check(assert_type(i0.dayofweek, IntegerIndex), IntegerIndex, int) |
541 | | - check(assert_type(i0.day_of_week, IntegerIndex), IntegerIndex, int) |
542 | | - check(assert_type(i0.weekday, IntegerIndex), IntegerIndex, int) |
543 | | - check(assert_type(i0.dayofyear, IntegerIndex), IntegerIndex, int) |
544 | | - check(assert_type(i0.day_of_year, IntegerIndex), IntegerIndex, int) |
545 | | - check(assert_type(i0.quarter, IntegerIndex), IntegerIndex, int) |
546 | | - check(assert_type(i0.daysinmonth, IntegerIndex), IntegerIndex, int) |
547 | | - check(assert_type(i0.days_in_month, IntegerIndex), IntegerIndex, int) |
| 534 | + check(assert_type(i0.year, NumericIndex), NumericIndex, int) |
| 535 | + check(assert_type(i0.month, NumericIndex), NumericIndex, int) |
| 536 | + check(assert_type(i0.day, NumericIndex), NumericIndex, int) |
| 537 | + check(assert_type(i0.hour, NumericIndex), NumericIndex, int) |
| 538 | + check(assert_type(i0.minute, NumericIndex), NumericIndex, int) |
| 539 | + check(assert_type(i0.second, NumericIndex), NumericIndex, int) |
| 540 | + check(assert_type(i0.dayofweek, NumericIndex), NumericIndex, int) |
| 541 | + check(assert_type(i0.day_of_week, NumericIndex), NumericIndex, int) |
| 542 | + check(assert_type(i0.weekday, NumericIndex), NumericIndex, int) |
| 543 | + check(assert_type(i0.dayofyear, NumericIndex), NumericIndex, int) |
| 544 | + check(assert_type(i0.day_of_year, NumericIndex), NumericIndex, int) |
| 545 | + check(assert_type(i0.quarter, NumericIndex), NumericIndex, int) |
| 546 | + check(assert_type(i0.daysinmonth, NumericIndex), NumericIndex, int) |
| 547 | + check(assert_type(i0.days_in_month, NumericIndex), NumericIndex, int) |
548 | 548 | check(assert_type(i0.freq, Optional[BaseOffset]), BaseOffset) |
549 | 549 | check(assert_type(i0.strftime("%Y"), pd.Index), pd.Index, str) |
550 | 550 | check(assert_type(i0.asfreq("D"), pd.PeriodIndex), pd.PeriodIndex, pd.Period) |
|
0 commit comments