diff --git a/pandas/tests/arrays/string_/test_string.py b/pandas/tests/arrays/string_/test_string.py index 0574061a6a544..89defb278d321 100644 --- a/pandas/tests/arrays/string_/test_string.py +++ b/pandas/tests/arrays/string_/test_string.py @@ -42,23 +42,16 @@ def cls(request): return request.param -def test_repr(dtype, request): - if dtype == "arrow_string": - reason = ( - "AssertionError: assert ' A\n0 a\n1 None\n2 b' " - "== ' A\n0 a\n1 \n2 b'" - ) - mark = pytest.mark.xfail(reason=reason) - request.node.add_marker(mark) - +def test_repr(dtype): df = pd.DataFrame({"A": pd.array(["a", pd.NA, "b"], dtype=dtype)}) expected = " A\n0 a\n1 \n2 b" assert repr(df) == expected - expected = "0 a\n1 \n2 b\nName: A, dtype: string" + expected = f"0 a\n1 \n2 b\nName: A, dtype: {dtype}" assert repr(df.A) == expected - expected = "\n['a', , 'b']\nLength: 3, dtype: string" + arr_name = "ArrowStringArray" if dtype == "arrow_string" else "StringArray" + expected = f"<{arr_name}>\n['a', , 'b']\nLength: 3, dtype: {dtype}" assert repr(df.A.array) == expected