Skip to content

Commit cab4cf4

Browse files
Backport PR #52929 on branch 2.0.x (TST/BUG: pyarrow test fixtures upcasting dtypes) (#52934)
Backport PR #52929: TST/BUG: pyarrow test fixtures upcasting dtypes Co-authored-by: Luke Manley <[email protected]>
1 parent a1c6fe0 commit cab4cf4

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
@@ -131,7 +131,7 @@ def data(dtype):
131131
@pytest.fixture
132132
def data_missing(data):
133133
"""Length-2 array with [NA, Valid]"""
134-
return type(data)._from_sequence([None, data[0]])
134+
return type(data)._from_sequence([None, data[0]], dtype=data.dtype)
135135

136136

137137
@pytest.fixture(params=["data", "data_missing"])
@@ -214,7 +214,8 @@ def data_for_sorting(data_for_grouping):
214214
A < B < C
215215
"""
216216
return type(data_for_grouping)._from_sequence(
217-
[data_for_grouping[0], data_for_grouping[7], data_for_grouping[4]]
217+
[data_for_grouping[0], data_for_grouping[7], data_for_grouping[4]],
218+
dtype=data_for_grouping.dtype,
218219
)
219220

220221

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

233235

0 commit comments

Comments
 (0)