Skip to content

Commit d35bce8

Browse files
committed
no-pyarrow fixup
1 parent 98b4c7a commit d35bce8

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

pandas/tests/series/indexing/test_setitem.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,12 @@
33
date,
44
datetime,
55
)
6+
from decimal import Decimal
67

78
import numpy as np
89
import pytest
910

11+
from pandas.compat import HAS_PYARROW
1012
from pandas.compat.numpy import np_version_gt2
1113
from pandas.errors import IndexingError
1214

@@ -588,7 +590,14 @@ def test_setitem_enlargement_object_none(self, nulls_fixture, using_infer_string
588590
ser = Series(["a", "b"])
589591
ser[3] = nulls_fixture
590592

591-
expected = Series(["a", "b", nulls_fixture], index=[0, 1, 3], dtype="str")
593+
dtype = (
594+
"str"
595+
if using_infer_string
596+
and not (isinstance(nulls_fixture, Decimal) and not HAS_PYARROW)
597+
else object
598+
)
599+
600+
expected = Series(["a", "b", nulls_fixture], index=[0, 1, 3], dtype=dtype)
592601
tm.assert_series_equal(ser, expected)
593602
if using_infer_string:
594603
ser[3] is np.nan

0 commit comments

Comments
 (0)