Skip to content

Commit 0093c03

Browse files
committed
fix import statements in test_dtypes
1 parent f26abfe commit 0093c03

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

pandas/tests/arrays/categorical/test_dtypes.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,10 @@
22
import pyarrow as pa
33
import pytest
44

5-
from pandas.core.dtypes.dtypes import CategoricalDtype
5+
from pandas.core.dtypes.dtypes import (
6+
ArrowDtype,
7+
CategoricalDtype,
8+
)
69

710
from pandas import (
811
Categorical,
@@ -12,7 +15,6 @@
1215
Series,
1316
Timestamp,
1417
)
15-
from pandas.core.dtypes.dtypes import ArrowDtype
1618
import pandas._testing as tm
1719

1820

@@ -141,14 +143,14 @@ def test_interval_index_category(self):
141143

142144
def test_values_is_index():
143145
# GH 60563
144-
values = Index(['a1', 'a2'], dtype=ArrowDtype(pa.string()))
146+
values = Index(["a1", "a2"], dtype=ArrowDtype(pa.string()))
145147
arr = values._data._pa_array.combine_chunks()
146148

147149
assert arr.equals(values._data._pa_array.combine_chunks())
148150

149151
def test_values_is_not_index():
150152
# GH 60563
151-
values = Series(['a1', 'a2'], dtype=ArrowDtype(pa.string()))
153+
values = Series(["a1", "a2"], dtype=ArrowDtype(pa.string()))
152154
arr = values._pa_array.combine_chunks()
153155

154-
assert arr.equals(values._pa_array.combine_chunks())
156+
assert arr.equals(values._pa_array.combine_chunks())

0 commit comments

Comments
 (0)