From 4426a44648b7eb6b993a2e3d24498763a5d90939 Mon Sep 17 00:00:00 2001 From: Luke Manley Date: Wed, 26 Apr 2023 11:53:48 -0400 Subject: [PATCH] Backport PR #52929: TST/BUG: pyarrow test fixtures upcasting dtypes --- 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 d557f5efc0a8a..bc642faeed163 100644 --- a/pandas/tests/extension/test_arrow.py +++ b/pandas/tests/extension/test_arrow.py @@ -131,7 +131,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"]) @@ -214,7 +214,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, ) @@ -227,7 +228,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, )