@@ -290,8 +290,10 @@ def test_subclass(self):
290
290
assert issubclass (type (a ), type (b ))
291
291
292
292
def test_compat (self , dtype ):
293
- assert is_datetime64tz_dtype (dtype )
294
- assert is_datetime64tz_dtype ("datetime64[ns, US/Eastern]" )
293
+ msg = "is_datetime64tz_dtype is deprecated"
294
+ with tm .assert_produces_warning (FutureWarning , match = msg ):
295
+ assert is_datetime64tz_dtype (dtype )
296
+ assert is_datetime64tz_dtype ("datetime64[ns, US/Eastern]" )
295
297
assert is_datetime64_any_dtype (dtype )
296
298
assert is_datetime64_any_dtype ("datetime64[ns, US/Eastern]" )
297
299
assert is_datetime64_ns_dtype (dtype )
@@ -349,25 +351,28 @@ def test_equality(self, dtype):
349
351
assert dtype == "M8[ns, US/Eastern]"
350
352
351
353
def test_basic (self , dtype ):
352
- assert is_datetime64tz_dtype (dtype )
354
+ msg = "is_datetime64tz_dtype is deprecated"
355
+ with tm .assert_produces_warning (FutureWarning , match = msg ):
356
+ assert is_datetime64tz_dtype (dtype )
353
357
354
358
dr = date_range ("20130101" , periods = 3 , tz = "US/Eastern" )
355
359
s = Series (dr , name = "A" )
356
360
357
361
# dtypes
358
- assert is_datetime64tz_dtype (s .dtype )
359
- assert is_datetime64tz_dtype (s )
360
- assert not is_datetime64tz_dtype (np .dtype ("float64" ))
361
- assert not is_datetime64tz_dtype (1.0 )
362
+ with tm .assert_produces_warning (FutureWarning , match = msg ):
363
+ assert is_datetime64tz_dtype (s .dtype )
364
+ assert is_datetime64tz_dtype (s )
365
+ assert not is_datetime64tz_dtype (np .dtype ("float64" ))
366
+ assert not is_datetime64tz_dtype (1.0 )
362
367
363
368
def test_dst (self ):
364
369
dr1 = date_range ("2013-01-01" , periods = 3 , tz = "US/Eastern" )
365
370
s1 = Series (dr1 , name = "A" )
366
- assert is_datetime64tz_dtype (s1 )
371
+ assert isinstance (s1 . dtype , DatetimeTZDtype )
367
372
368
373
dr2 = date_range ("2013-08-01" , periods = 3 , tz = "US/Eastern" )
369
374
s2 = Series (dr2 , name = "A" )
370
- assert is_datetime64tz_dtype (s2 )
375
+ assert isinstance (s2 . dtype , DatetimeTZDtype )
371
376
assert s1 .dtype == s2 .dtype
372
377
373
378
@pytest .mark .parametrize ("tz" , ["UTC" , "US/Eastern" ])
@@ -595,7 +600,9 @@ def test_hash_vs_equality(self, dtype):
595
600
def test_construction (self , subtype ):
596
601
i = IntervalDtype (subtype , closed = "right" )
597
602
assert i .subtype == np .dtype ("int64" )
598
- assert is_interval_dtype (i )
603
+ msg = "is_interval_dtype is deprecated"
604
+ with tm .assert_produces_warning (FutureWarning , match = msg ):
605
+ assert is_interval_dtype (i )
599
606
600
607
@pytest .mark .parametrize (
601
608
"subtype" , ["interval[int64]" , "Interval[int64]" , "int64" , np .dtype ("int64" )]
@@ -616,7 +623,9 @@ def test_construction_generic(self, subtype):
616
623
# generic
617
624
i = IntervalDtype (subtype )
618
625
assert i .subtype is None
619
- assert is_interval_dtype (i )
626
+ msg = "is_interval_dtype is deprecated"
627
+ with tm .assert_produces_warning (FutureWarning , match = msg ):
628
+ assert is_interval_dtype (i )
620
629
621
630
@pytest .mark .parametrize (
622
631
"subtype" ,
@@ -787,31 +796,35 @@ def test_name_repr_generic(self, subtype):
787
796
assert dtype .name == "interval"
788
797
789
798
def test_basic (self , dtype ):
790
- assert is_interval_dtype (dtype )
799
+ msg = "is_interval_dtype is deprecated"
800
+ with tm .assert_produces_warning (FutureWarning , match = msg ):
801
+ assert is_interval_dtype (dtype )
791
802
792
- ii = IntervalIndex .from_breaks (range (3 ))
803
+ ii = IntervalIndex .from_breaks (range (3 ))
793
804
794
- assert is_interval_dtype (ii .dtype )
795
- assert is_interval_dtype (ii )
805
+ assert is_interval_dtype (ii .dtype )
806
+ assert is_interval_dtype (ii )
796
807
797
- s = Series (ii , name = "A" )
808
+ s = Series (ii , name = "A" )
798
809
799
- assert is_interval_dtype (s .dtype )
800
- assert is_interval_dtype (s )
810
+ assert is_interval_dtype (s .dtype )
811
+ assert is_interval_dtype (s )
801
812
802
813
def test_basic_dtype (self ):
803
- assert is_interval_dtype ("interval[int64, both]" )
804
- assert is_interval_dtype (IntervalIndex .from_tuples ([(0 , 1 )]))
805
- assert is_interval_dtype (IntervalIndex .from_breaks (np .arange (4 )))
806
- assert is_interval_dtype (
807
- IntervalIndex .from_breaks (date_range ("20130101" , periods = 3 ))
808
- )
809
- assert not is_interval_dtype ("U" )
810
- assert not is_interval_dtype ("S" )
811
- assert not is_interval_dtype ("foo" )
812
- assert not is_interval_dtype (np .object_ )
813
- assert not is_interval_dtype (np .int64 )
814
- assert not is_interval_dtype (np .float64 )
814
+ msg = "is_interval_dtype is deprecated"
815
+ with tm .assert_produces_warning (FutureWarning , match = msg ):
816
+ assert is_interval_dtype ("interval[int64, both]" )
817
+ assert is_interval_dtype (IntervalIndex .from_tuples ([(0 , 1 )]))
818
+ assert is_interval_dtype (IntervalIndex .from_breaks (np .arange (4 )))
819
+ assert is_interval_dtype (
820
+ IntervalIndex .from_breaks (date_range ("20130101" , periods = 3 ))
821
+ )
822
+ assert not is_interval_dtype ("U" )
823
+ assert not is_interval_dtype ("S" )
824
+ assert not is_interval_dtype ("foo" )
825
+ assert not is_interval_dtype (np .object_ )
826
+ assert not is_interval_dtype (np .int64 )
827
+ assert not is_interval_dtype (np .float64 )
815
828
816
829
def test_caching (self ):
817
830
IntervalDtype .reset_cache ()
@@ -1113,9 +1126,14 @@ def test_is_dtype_no_warning(check):
1113
1126
data = pd .DataFrame ({"A" : [1 , 2 ]})
1114
1127
1115
1128
warn = None
1116
- msg = "is_categorical_dtype is deprecated"
1117
- if check is is_categorical_dtype :
1129
+ msg = f"{ check .__name__ } is deprecated"
1130
+ if (
1131
+ check is is_categorical_dtype
1132
+ or check is is_interval_dtype
1133
+ or check is is_datetime64tz_dtype
1134
+ ):
1118
1135
warn = FutureWarning
1136
+
1119
1137
with tm .assert_produces_warning (warn , match = msg ):
1120
1138
check (data )
1121
1139
0 commit comments