|
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