1
1
import pytest
2
2
3
+ from pandas .core .dtypes .common import is_extension_array_dtype
4
+
3
5
import pandas as pd
4
6
from pandas import (
5
7
Categorical ,
@@ -105,7 +107,7 @@ def test_series_not_equal_metadata_mismatch(kwargs):
105
107
106
108
107
109
@pytest .mark .parametrize ("data1,data2" , [(0.12345 , 0.12346 ), (0.1235 , 0.1236 )])
108
- @pytest .mark .parametrize ("dtype" , ["float32" , "float64" ])
110
+ @pytest .mark .parametrize ("dtype" , ["float32" , "float64" , "Float32" ])
109
111
@pytest .mark .parametrize ("decimals" , [0 , 1 , 2 , 3 , 5 , 10 ])
110
112
def test_less_precise (data1 , data2 , dtype , decimals ):
111
113
rtol = 10 ** - decimals
@@ -115,7 +117,10 @@ def test_less_precise(data1, data2, dtype, decimals):
115
117
if (decimals == 5 or decimals == 10 ) or (
116
118
decimals >= 3 and abs (data1 - data2 ) >= 0.0005
117
119
):
118
- msg = "Series values are different"
120
+ if is_extension_array_dtype (dtype ):
121
+ msg = "ExtensionArray are different"
122
+ else :
123
+ msg = "Series values are different"
119
124
with pytest .raises (AssertionError , match = msg ):
120
125
tm .assert_series_equal (s1 , s2 , rtol = rtol )
121
126
else :
0 commit comments