31 lines
789 B
Python
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)
|