|
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