Skip to content

Commit 52ddb2e

Browse files
authored
Adjust tests in xml folder for new string option (#56198)
1 parent 762b61d commit 52ddb2e

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

pandas/tests/io/xml/test_xml.py

+10-2
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
ArrowStringArray,
3333
StringArray,
3434
)
35+
from pandas.core.arrays.string_arrow import ArrowStringArrayNumpySemantics
3536

3637
from pandas.io.common import get_handle
3738
from pandas.io.xml import read_xml
@@ -2004,7 +2005,9 @@ def test_s3_parser_consistency(s3_public_bucket_with_data, s3so):
20042005
tm.assert_frame_equal(df_lxml, df_etree)
20052006

20062007

2007-
def test_read_xml_nullable_dtypes(parser, string_storage, dtype_backend):
2008+
def test_read_xml_nullable_dtypes(
2009+
parser, string_storage, dtype_backend, using_infer_string
2010+
):
20082011
# GH#50500
20092012
data = """<?xml version='1.0' encoding='utf-8'?>
20102013
<data xmlns="http://example.com">
@@ -2032,7 +2035,12 @@ def test_read_xml_nullable_dtypes(parser, string_storage, dtype_backend):
20322035
</row>
20332036
</data>"""
20342037

2035-
if string_storage == "python":
2038+
if using_infer_string:
2039+
pa = pytest.importorskip("pyarrow")
2040+
string_array = ArrowStringArrayNumpySemantics(pa.array(["x", "y"]))
2041+
string_array_na = ArrowStringArrayNumpySemantics(pa.array(["x", None]))
2042+
2043+
elif string_storage == "python":
20362044
string_array = StringArray(np.array(["x", "y"], dtype=np.object_))
20372045
string_array_na = StringArray(np.array(["x", NA], dtype=np.object_))
20382046

0 commit comments

Comments
 (0)