File tree Expand file tree Collapse file tree 2 files changed +8
-1
lines changed Expand file tree Collapse file tree 2 files changed +8
-1
lines changed Original file line number Diff line number Diff line change 14
14
pa_version_under13p0 = _palv < Version ("13.0.0" )
15
15
pa_version_under14p0 = _palv < Version ("14.0.0" )
16
16
pa_version_under14p1 = _palv < Version ("14.0.1" )
17
+ pa_version_under15p0 = _palv < Version ("15.0.0" )
17
18
except ImportError :
18
19
pa_version_under10p1 = True
19
20
pa_version_under11p0 = True
20
21
pa_version_under12p0 = True
21
22
pa_version_under13p0 = True
22
23
pa_version_under14p0 = True
23
24
pa_version_under14p1 = True
25
+ pa_version_under15p0 = True
Original file line number Diff line number Diff line change 15
15
from pandas .compat .pyarrow import (
16
16
pa_version_under11p0 ,
17
17
pa_version_under13p0 ,
18
+ pa_version_under15p0 ,
18
19
)
19
20
20
21
import pandas as pd
@@ -752,7 +753,10 @@ def test_unsupported_float16(self, pa):
752
753
# Not able to write float 16 column using pyarrow.
753
754
data = np .arange (2 , 10 , dtype = np .float16 )
754
755
df = pd .DataFrame (data = data , columns = ["fp16" ])
755
- self .check_external_error_on_write (df , pa , pyarrow .ArrowException )
756
+ if pa_version_under15p0 :
757
+ self .check_external_error_on_write (df , pa , pyarrow .ArrowException )
758
+ else :
759
+ check_round_trip (df , pa )
756
760
757
761
@pytest .mark .xfail (
758
762
is_platform_windows (),
@@ -761,6 +765,7 @@ def test_unsupported_float16(self, pa):
761
765
"dtypes are passed to_parquet function in windows"
762
766
),
763
767
)
768
+ @pytest .mark .skipif (not pa_version_under15p0 , reason = "float16 works on 15" )
764
769
@pytest .mark .parametrize ("path_type" , [str , pathlib .Path ])
765
770
def test_unsupported_float16_cleanup (self , pa , path_type ):
766
771
# #44847, #44914
You can’t perform that action at this time.
0 commit comments