|
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