3RNN/Lib/site-packages/pandas/tests/util/test_shares_memory.py
2024-05-26 19:49:15 +02:00

31 lines
789 B
Python

import pandas.util._test_decorators as td
import pandas as pd
import pandas._testing as tm
def test_shares_memory_interval():
obj = pd.interval_range(1, 5)
assert tm.shares_memory(obj, obj)
assert tm.shares_memory(obj, obj._data)
assert tm.shares_memory(obj, obj[::-1])
assert tm.shares_memory(obj, obj[:2])
assert not tm.shares_memory(obj, obj._data.copy())
@td.skip_if_no("pyarrow")
def test_shares_memory_string():
# GH#55823
import pyarrow as pa
obj = pd.array(["a", "b"], dtype="string[pyarrow]")
assert tm.shares_memory(obj, obj)
obj = pd.array(["a", "b"], dtype="string[pyarrow_numpy]")
assert tm.shares_memory(obj, obj)
obj = pd.array(["a", "b"], dtype=pd.ArrowDtype(pa.string()))
assert tm.shares_memory(obj, obj)