Skip to content

Commit c632485

Browse files
authored
TST/BUG: pyarrow test fixtures upcasting dtypes (#52929)
fix arrow test fixtures upcasting
1 parent b877b43 commit c632485

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

pandas/tests/extension/test_arrow.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def data(dtype):
130130
@pytest.fixture
131131
def data_missing(data):
132132
"""Length-2 array with [NA, Valid]"""
133-
return type(data)._from_sequence([None, data[0]])
133+
return type(data)._from_sequence([None, data[0]], dtype=data.dtype)
134134

135135

136136
@pytest.fixture(params=["data", "data_missing"])
@@ -213,7 +213,8 @@ def data_for_sorting(data_for_grouping):
213213
A < B < C
214214
"""
215215
return type(data_for_grouping)._from_sequence(
216-
[data_for_grouping[0], data_for_grouping[7], data_for_grouping[4]]
216+
[data_for_grouping[0], data_for_grouping[7], data_for_grouping[4]],
217+
dtype=data_for_grouping.dtype,
217218
)
218219

219220

@@ -226,7 +227,8 @@ def data_missing_for_sorting(data_for_grouping):
226227
A < B and NA missing.
227228
"""
228229
return type(data_for_grouping)._from_sequence(
229-
[data_for_grouping[0], data_for_grouping[2], data_for_grouping[4]]
230+
[data_for_grouping[0], data_for_grouping[2], data_for_grouping[4]],
231+
dtype=data_for_grouping.dtype,
230232
)
231233

232234

0 commit comments

Comments
 (0)