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