From 13da24b21f10d65d8494d4ed776f696ec7009707 Mon Sep 17 00:00:00 2001 From: Luke Manley Date: Wed, 26 Apr 2023 07:25:25 -0400 Subject: [PATCH] fix arrow test fixtures upcasting --- pandas/tests/extension/test_arrow.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/pandas/tests/extension/test_arrow.py b/pandas/tests/extension/test_arrow.py index 0300b271acc3f..e9a9dbbc8d965 100644 --- a/pandas/tests/extension/test_arrow.py +++ b/pandas/tests/extension/test_arrow.py @@ -130,7 +130,7 @@ def data(dtype): @pytest.fixture def data_missing(data): """Length-2 array with [NA, Valid]""" - return type(data)._from_sequence([None, data[0]]) + return type(data)._from_sequence([None, data[0]], dtype=data.dtype) @pytest.fixture(params=["data", "data_missing"]) @@ -213,7 +213,8 @@ def data_for_sorting(data_for_grouping): A < B < C """ return type(data_for_grouping)._from_sequence( - [data_for_grouping[0], data_for_grouping[7], data_for_grouping[4]] + [data_for_grouping[0], data_for_grouping[7], data_for_grouping[4]], + dtype=data_for_grouping.dtype, ) @@ -226,7 +227,8 @@ def data_missing_for_sorting(data_for_grouping): A < B and NA missing. """ return type(data_for_grouping)._from_sequence( - [data_for_grouping[0], data_for_grouping[2], data_for_grouping[4]] + [data_for_grouping[0], data_for_grouping[2], data_for_grouping[4]], + dtype=data_for_grouping.dtype, )