added additional utility functions for easier access to types

Author: Chaluvadi

tests/utility_functions.py

@@ -1,7 +1,7 @@
 import pytest
 
 import arrayfire_wrapper.lib as wrapper
-from arrayfire_wrapper.dtypes import Dtype, c64, f16, f64
+from arrayfire_wrapper.dtypes import Dtype, c32, c64, f16, f32, f64, s16, s32, s64, u8, u16, u32, u64
 
 
 def check_type_supported(dtype: Dtype) -> None:
@@ -11,3 +11,18 @@ def check_type_supported(dtype: Dtype) -> None:
 
     if dtype == f16 and not wrapper.get_half_support():
         pytest.skip("Device does not support half types.")
+
+
+def get_complex_types() -> list:
+    """Returns all complex types"""
+    return [c32, c64]
+
+
+def get_real_types() -> list:
+    """Returns all real types"""
+    return [s16, s32, s64, u8, u16, u32, u64, f16, f32, f64]
+
+
+def get_all_types() -> list:
+    """Returns all types"""
+    return [s16, s32, s64, u8, u16, u32, u64, f16, f32, f64, c32, c64]