Skip to content

Commit 5ce3c14

Browse files
author
Chaluvadi
committed
Fixed utility functions
1 parent 8bc8102 commit 5ce3c14

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

tests/test_utilities.py

+3-9
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,13 @@
1-
import arrayfire_wrapper.lib as wrapper
21
import pytest
32

4-
from arrayfire_wrapper.dtypes import f64, f16, c32, c64, Dtype
5-
6-
7-
def is_cmplx_type(dtype: Dtype) -> bool:
8-
"""Checks to see if the specified type is a complex type"""
9-
return dtype == c32 or dtype == c64
3+
import arrayfire_wrapper.lib as wrapper
4+
from arrayfire_wrapper.dtypes import Dtype, c64, f16, f64
105

116

127
def check_type_supported(dtype: Dtype) -> None:
138
"""Checks to see if the specified type is supported by the current system"""
149
if dtype in [f64, c64] and not wrapper.get_dbl_support():
15-
pytest.skip("Device does not support doubel types.")
10+
pytest.skip("Device does not support double types")
1611

1712
if dtype == f16 and not wrapper.get_half_support():
1813
pytest.skip("Device does not support half types.")
19-

0 commit comments

Comments
 (0)