|
1 | 1 | from datetime import date, datetime, time as dt_time, timedelta
|
2 |
| -from typing import Type |
| 2 | +from typing import Dict, List, Tuple, Type |
3 | 3 |
|
4 | 4 | import numpy as np
|
5 | 5 | import pytest
|
|
31 | 31 | import pandas.tseries.offsets as offsets
|
32 | 32 | from pandas.tseries.offsets import (
|
33 | 33 | FY5253,
|
| 34 | + BaseOffset, |
34 | 35 | BDay,
|
35 | 36 | BMonthBegin,
|
36 | 37 | BMonthEnd,
|
@@ -90,6 +91,7 @@ def test_to_M8():
|
90 | 91 | #####
|
91 | 92 | # DateOffset Tests
|
92 | 93 | #####
|
| 94 | +_ApplyCases = List[Tuple[BaseOffset, Dict[datetime, datetime]]] |
93 | 95 |
|
94 | 96 |
|
95 | 97 | class Base:
|
@@ -741,7 +743,7 @@ def test_onOffset(self):
|
741 | 743 | for offset, d, expected in tests:
|
742 | 744 | assert_onOffset(offset, d, expected)
|
743 | 745 |
|
744 |
| - apply_cases = [] |
| 746 | + apply_cases = [] # type: _ApplyCases |
745 | 747 | apply_cases.append(
|
746 | 748 | (
|
747 | 749 | BDay(),
|
@@ -2629,7 +2631,7 @@ def test_onOffset(self, case):
|
2629 | 2631 | offset, d, expected = case
|
2630 | 2632 | assert_onOffset(offset, d, expected)
|
2631 | 2633 |
|
2632 |
| - apply_cases = [] |
| 2634 | + apply_cases = [] # type: _ApplyCases |
2633 | 2635 | apply_cases.append(
|
2634 | 2636 | (
|
2635 | 2637 | CDay(),
|
@@ -2876,7 +2878,7 @@ def test_onOffset(self, case):
|
2876 | 2878 | offset, d, expected = case
|
2877 | 2879 | assert_onOffset(offset, d, expected)
|
2878 | 2880 |
|
2879 |
| - apply_cases = [] |
| 2881 | + apply_cases = [] # type: _ApplyCases |
2880 | 2882 | apply_cases.append(
|
2881 | 2883 | (
|
2882 | 2884 | CBMonthEnd(),
|
@@ -3025,7 +3027,7 @@ def test_onOffset(self, case):
|
3025 | 3027 | offset, dt, expected = case
|
3026 | 3028 | assert_onOffset(offset, dt, expected)
|
3027 | 3029 |
|
3028 |
| - apply_cases = [] |
| 3030 | + apply_cases = [] # type: _ApplyCases |
3029 | 3031 | apply_cases.append(
|
3030 | 3032 | (
|
3031 | 3033 | CBMonthBegin(),
|
|
0 commit comments