Skip to content

Commit 27c0a3c

Browse files
AzeezIshsyurkevi
AzeezIsh
authored andcommitted
Removed unneeded comments, fixed import statement.
1 parent b6636d7 commit 27c0a3c

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

tests/test_muldiv.py

+5-8
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,7 @@
66
import arrayfire_wrapper.dtypes as dtype
77
import arrayfire_wrapper.lib as wrapper
88

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
1310

1411
@pytest.mark.parametrize(
1512
"shape",
@@ -59,10 +56,10 @@ def test_multiply_negative_shapes() -> None:
5956
), f"Failed for shapes {lhs_shape} and {rhs_shape}"
6057

6158

62-
@pytest.mark.parametrize("dtype_name", util.get_all_types())
59+
@pytest.mark.parametrize("dtype_name", get_all_types())
6360
def test_multiply_supported_dtypes(dtype_name: dtype.Dtype) -> None:
6461
"""Test multiplication operation across all supported data types."""
65-
util.check_type_supported(dtype_name)
62+
check_type_supported(dtype_name)
6663
shape = (5, 5)
6764
lhs = wrapper.randu(shape, dtype_name)
6865
rhs = wrapper.randu(shape, dtype_name)
@@ -174,10 +171,10 @@ def test_divide_negative_shapes() -> None:
174171
), f"Failed for shapes {lhs_shape} and {rhs_shape}"
175172

176173

177-
@pytest.mark.parametrize("dtype_name", util.get_all_types())
174+
@pytest.mark.parametrize("dtype_name", get_all_types())
178175
def test_divide_supported_dtypes(dtype_name: dtype.Dtype) -> None:
179176
"""Test division operation across all supported data types."""
180-
util.check_type_supported(dtype_name)
177+
check_type_supported(dtype_name)
181178
shape = (5, 5)
182179
lhs = wrapper.randu(shape, dtype_name)
183180
rhs = wrapper.randu(shape, dtype_name)

0 commit comments

Comments
 (0)