CLN: Fix typing in pandas\tests\arrays\test_datetimelike.py (pandas-d…

…ev#28926) (pandas-dev#29014)
proost · Dec 19, 2019 · 523af80 · 523af80
1 parent 4ffaa39
commit 523af80
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/pandas/tests/arrays/test_datetimelike.py b/pandas/tests/arrays/test_datetimelike.py
@@ -1,10 +1,15 @@
+from typing import Type, Union
+
 import numpy as np
 import pytest
 
 from pandas._libs import OutOfBoundsDatetime
 
 import pandas as pd
 from pandas.core.arrays import DatetimeArray, PeriodArray, TimedeltaArray
+from pandas.core.indexes.datetimes import DatetimeIndex
+from pandas.core.indexes.period import PeriodIndex
+from pandas.core.indexes.timedeltas import TimedeltaIndex
 import pandas.util.testing as tm
 
 
@@ -52,7 +57,7 @@ def timedelta_index(request):
 
 
 class SharedTests:
-    index_cls = None
+    index_cls = None  # type: Type[Union[DatetimeIndex, PeriodIndex, TimedeltaIndex]]
 
     def test_compare_len1_raises(self):
         # make sure we raise when comparing with different lengths, specific

diff --git a/setup.cfg b/setup.cfg
@@ -136,9 +136,6 @@ ignore_errors=True
 [mypy-pandas.tests.arithmetic.test_datetime64]
 ignore_errors=True
 
-[mypy-pandas.tests.arrays.test_datetimelike]
-ignore_errors=True
-
 [mypy-pandas.tests.dtypes.test_common]
 ignore_errors=True