Skip to content

Commit 823d888

Browse files
simonjayhawkinsvladu
authored andcommitted
TST: [ArrowStringArray] remove xfail from test_repr (pandas-dev#40678)
1 parent dbe2883 commit 823d888

File tree

1 file changed

+4
-11
lines changed

1 file changed

+4
-11
lines changed

pandas/tests/arrays/string_/test_string.py

+4-11
Original file line numberDiff line numberDiff line change
@@ -42,23 +42,16 @@ def cls(request):
4242
return request.param
4343

4444

45-
def test_repr(dtype, request):
46-
if dtype == "arrow_string":
47-
reason = (
48-
"AssertionError: assert ' A\n0 a\n1 None\n2 b' "
49-
"== ' A\n0 a\n1 <NA>\n2 b'"
50-
)
51-
mark = pytest.mark.xfail(reason=reason)
52-
request.node.add_marker(mark)
53-
45+
def test_repr(dtype):
5446
df = pd.DataFrame({"A": pd.array(["a", pd.NA, "b"], dtype=dtype)})
5547
expected = " A\n0 a\n1 <NA>\n2 b"
5648
assert repr(df) == expected
5749

58-
expected = "0 a\n1 <NA>\n2 b\nName: A, dtype: string"
50+
expected = f"0 a\n1 <NA>\n2 b\nName: A, dtype: {dtype}"
5951
assert repr(df.A) == expected
6052

61-
expected = "<StringArray>\n['a', <NA>, 'b']\nLength: 3, dtype: string"
53+
arr_name = "ArrowStringArray" if dtype == "arrow_string" else "StringArray"
54+
expected = f"<{arr_name}>\n['a', <NA>, 'b']\nLength: 3, dtype: {dtype}"
6255
assert repr(df.A.array) == expected
6356

6457

0 commit comments

Comments
 (0)