@@ -528,6 +528,7 @@ def test_groupby_extension_transform(self, data_for_grouping, request):
528
528
def test_groupby_extension_apply (
529
529
self , data_for_grouping , groupby_apply_op , request
530
530
):
531
+ pa_dtype = data_for_grouping .dtype .pyarrow_dtype
531
532
with tm .maybe_produces_warning (
532
533
PerformanceWarning ,
533
534
pa_version_under7p0 and not pa .types .is_duration (pa_dtype ),
@@ -770,6 +771,7 @@ def test_value_counts(self, all_data, dropna, request):
770
771
super ().test_value_counts (all_data , dropna )
771
772
772
773
def test_value_counts_with_normalize (self , data , request ):
774
+ pa_dtype = data .dtype .pyarrow_dtype
773
775
with tm .maybe_produces_warning (
774
776
PerformanceWarning ,
775
777
pa_version_under7p0 and not pa .types .is_duration (pa_dtype ),
@@ -869,6 +871,7 @@ def test_sort_values_missing(
869
871
870
872
@pytest .mark .parametrize ("ascending" , [True , False ])
871
873
def test_sort_values_frame (self , data_for_sorting , ascending , request ):
874
+ pa_dtype = data_for_sorting .dtype .pyarrow_dtype
872
875
with tm .maybe_produces_warning (
873
876
PerformanceWarning ,
874
877
pa_version_under7p0 and not pa .types .is_duration (pa_dtype ),
0 commit comments