16
16
assert_dtype_allclose ,
17
17
get_all_dtypes ,
18
18
get_complex_dtypes ,
19
+ get_float_complex_dtypes ,
19
20
get_float_dtypes ,
20
21
has_support_aspect64 ,
21
22
is_cpu_device ,
@@ -966,7 +967,7 @@ def test_invalid_out(self, out):
966
967
967
968
968
969
class TestDivide :
969
- @pytest .mark .parametrize ("dtype" , get_float_dtypes () + get_complex_dtypes ())
970
+ @pytest .mark .parametrize ("dtype" , get_float_complex_dtypes ())
970
971
def test_divide (self , dtype ):
971
972
array1_data = numpy .arange (10 )
972
973
array2_data = numpy .arange (5 , 15 )
@@ -983,12 +984,11 @@ def test_divide(self, dtype):
983
984
np_array2 = numpy .array (array2_data , dtype = dtype )
984
985
expected = numpy .divide (np_array1 , np_array2 , out = out )
985
986
986
- tol = 1e-07
987
- assert_allclose (expected , result , rtol = tol , atol = tol )
988
- assert_allclose (out , dp_out , rtol = tol , atol = tol )
987
+ assert_dtype_allclose (result , expected )
988
+ assert_dtype_allclose (dp_out , out )
989
989
990
990
@pytest .mark .usefixtures ("suppress_divide_invalid_numpy_warnings" )
991
- @pytest .mark .parametrize ("dtype" , get_float_dtypes () + get_complex_dtypes ())
991
+ @pytest .mark .parametrize ("dtype" , get_float_complex_dtypes ())
992
992
def test_out_dtypes (self , dtype ):
993
993
size = 10
994
994
@@ -1010,12 +1010,10 @@ def test_out_dtypes(self, dtype):
1010
1010
dp_out = dpnp .empty (size , dtype = dtype )
1011
1011
1012
1012
result = dpnp .divide (dp_array1 , dp_array2 , out = dp_out )
1013
-
1014
- tol = 1e-07
1015
- assert_allclose (expected , result , rtol = tol , atol = tol )
1013
+ assert_dtype_allclose (result , expected )
1016
1014
1017
1015
@pytest .mark .usefixtures ("suppress_divide_invalid_numpy_warnings" )
1018
- @pytest .mark .parametrize ("dtype" , get_float_dtypes () + get_complex_dtypes ())
1016
+ @pytest .mark .parametrize ("dtype" , get_float_complex_dtypes ())
1019
1017
def test_out_overlap (self , dtype ):
1020
1018
size = 15
1021
1019
# DPNP
@@ -1026,10 +1024,9 @@ def test_out_overlap(self, dtype):
1026
1024
np_a = numpy .arange (2 * size , dtype = dtype )
1027
1025
numpy .divide (np_a [size ::], np_a [::2 ], out = np_a [:size :])
1028
1026
1029
- tol = 1e-07
1030
- assert_allclose (np_a , dp_a , rtol = tol , atol = tol )
1027
+ assert_dtype_allclose (dp_a , np_a )
1031
1028
1032
- @pytest .mark .parametrize ("dtype" , get_float_dtypes () + get_complex_dtypes ())
1029
+ @pytest .mark .parametrize ("dtype" , get_float_complex_dtypes ())
1033
1030
def test_inplace_strided_out (self , dtype ):
1034
1031
size = 21
1035
1032
0 commit comments