3RNN/Lib/site-packages/pandas/tests/strings/__init__.py

16 lines
367 B
Python
Raw Permalink Normal View History

2024-05-26 19:49:15 +02:00
import numpy as np
import pandas as pd
object_pyarrow_numpy = ("object", "string[pyarrow_numpy]")
def _convert_na_value(ser, expected):
if ser.dtype != object:
if ser.dtype.storage == "pyarrow_numpy":
expected = expected.fillna(np.nan)
else:
# GH#18463
expected = expected.fillna(pd.NA)
return expected