File tree 2 files changed +11
-5
lines changed
2 files changed +11
-5
lines changed Original file line number Diff line number Diff line change 13
13
pa_version_under9p0 = _palv < Version ("9.0.0" )
14
14
pa_version_under10p0 = _palv < Version ("10.0.0" )
15
15
pa_version_under11p0 = _palv < Version ("11.0.0" )
16
+ pa_version_under12p0 = _palv < Version ("12.0.0" )
16
17
pa_version_under13p0 = _palv < Version ("13.0.0" )
17
18
except ImportError :
18
19
pa_version_under7p0 = True
19
20
pa_version_under8p0 = True
20
21
pa_version_under9p0 = True
21
22
pa_version_under10p0 = True
22
23
pa_version_under11p0 = True
24
+ pa_version_under12p0 = True
23
25
pa_version_under13p0 = True
Original file line number Diff line number Diff line change 2
2
import pytest
3
3
4
4
from pandas .compat import pa_version_under7p0
5
+ from pandas .compat .pyarrow import pa_version_under12p0
5
6
import pandas .util ._test_decorators as td
6
7
7
8
import pandas as pd
@@ -200,11 +201,14 @@ def test_astype_arrow_timestamp(using_copy_on_write):
200
201
result = df .astype ("timestamp[ns][pyarrow]" )
201
202
if using_copy_on_write :
202
203
assert not result ._mgr ._has_no_reference (0 )
203
- # TODO(CoW): arrow is not setting copy=False in the Series constructor
204
- # under the hood
205
- assert not np .shares_memory (
206
- get_array (df , "a" ), get_array (result , "a" )._pa_array
207
- )
204
+ if pa_version_under12p0 :
205
+ assert not np .shares_memory (
206
+ get_array (df , "a" ), get_array (result , "a" )._pa_array
207
+ )
208
+ else :
209
+ assert np .shares_memory (
210
+ get_array (df , "a" ), get_array (result , "a" )._pa_array
211
+ )
208
212
209
213
210
214
def test_convert_dtypes_infer_objects (using_copy_on_write ):
You can’t perform that action at this time.
0 commit comments