|
6 | 6 | import arrayfire_wrapper.dtypes as dtype |
7 | 7 | import arrayfire_wrapper.lib as wrapper |
8 | 8 |
|
9 | | -# import arrayfire_wrapper.lib.mathematical_functions as ops |
10 | | - |
11 | | -from . import utility_functions as util |
12 | | - |
| 9 | +from utility_functions import check_type_supported, get_all_types |
13 | 10 |
|
14 | 11 | @pytest.mark.parametrize( |
15 | 12 | "shape", |
@@ -59,10 +56,10 @@ def test_multiply_negative_shapes() -> None: |
59 | 56 | ), f"Failed for shapes {lhs_shape} and {rhs_shape}" |
60 | 57 |
|
61 | 58 |
|
62 | | -@pytest.mark.parametrize("dtype_name", util.get_all_types()) |
| 59 | +@pytest.mark.parametrize("dtype_name", get_all_types()) |
63 | 60 | def test_multiply_supported_dtypes(dtype_name: dtype.Dtype) -> None: |
64 | 61 | """Test multiplication operation across all supported data types.""" |
65 | | - util.check_type_supported(dtype_name) |
| 62 | + check_type_supported(dtype_name) |
66 | 63 | shape = (5, 5) |
67 | 64 | lhs = wrapper.randu(shape, dtype_name) |
68 | 65 | rhs = wrapper.randu(shape, dtype_name) |
@@ -174,10 +171,10 @@ def test_divide_negative_shapes() -> None: |
174 | 171 | ), f"Failed for shapes {lhs_shape} and {rhs_shape}" |
175 | 172 |
|
176 | 173 |
|
177 | | -@pytest.mark.parametrize("dtype_name", util.get_all_types()) |
| 174 | +@pytest.mark.parametrize("dtype_name", get_all_types()) |
178 | 175 | def test_divide_supported_dtypes(dtype_name: dtype.Dtype) -> None: |
179 | 176 | """Test division operation across all supported data types.""" |
180 | | - util.check_type_supported(dtype_name) |
| 177 | + check_type_supported(dtype_name) |
181 | 178 | shape = (5, 5) |
182 | 179 | lhs = wrapper.randu(shape, dtype_name) |
183 | 180 | rhs = wrapper.randu(shape, dtype_name) |
|
0 commit comments