UPdated unit tests

Author: Diptorup Deb

numba_dpex/tests/experimental/IntEnumLiteral/test_compilation.py

@@ -2,12 +2,10 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-import dpctl
 import dpnp
-from numba.core import types
 
 import numba_dpex.experimental as exp_dpex
-from numba_dpex import DpctlSyclQueue, DpnpNdArray, Range, int64
+from numba_dpex import Range
 from numba_dpex.experimental.flag_enum import FlagEnum
 
 
@@ -36,24 +34,3 @@ def test_compilation_of_flag_enum():
     assert a[1] == MockFlags.FLAG2
     for idx in range(2, 9):
         assert a[idx] == 1
-
-
-def test_compilation_as_literal_constant():
-    @exp_dpex.device_func
-    def bitwise_or_flags(flag1, flag2):
-        return flag1 | flag2
-
-    def pass_flags_to_func(a):
-        f1 = MockFlags.FLAG1
-        f2 = MockFlags.FLAG2
-        a[0] = bitwise_or_flags(f1, f2)
-
-    queue_ty = DpctlSyclQueue(dpctl.SyclQueue())
-    i64arr_ty = DpnpNdArray(ndim=1, dtype=int64, layout="C", queue=queue_ty)
-    kernel_sig = types.void(i64arr_ty)
-
-    disp = exp_dpex.kernel(pass_flags_to_func)
-    disp.compile(kernel_sig)
-    kcres = disp.overloads[kernel_sig.args]
-    llvm_ir_mod = kcres.library._final_module
-    print(llvm_ir_mod)

numba_dpex/tests/experimental/codegen/test_intenum_literal_codegen.py

@@ -0,0 +1,44 @@
+# SPDX-FileCopyrightText: 2023 Intel Corporation
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import re
+
+import dpctl
+from numba.core import types
+
+import numba_dpex.experimental as exp_dpex
+from numba_dpex import DpctlSyclQueue, DpnpNdArray, int64
+from numba_dpex.experimental.flag_enum import FlagEnum
+
+
+class MockFlags(FlagEnum):
+    FLAG1 = 1
+    FLAG2 = 2
+
+
+def test_compilation_as_literal_constant():
+    @exp_dpex.device_func
+    def bitwise_or_flags(flag1, flag2):
+        return flag1 | flag2
+
+    def pass_flags_to_func(a):
+        f1 = MockFlags.FLAG1
+        f2 = MockFlags.FLAG2
+        a[0] = bitwise_or_flags(f1, f2)
+
+    queue_ty = DpctlSyclQueue(dpctl.SyclQueue())
+    i64arr_ty = DpnpNdArray(ndim=1, dtype=int64, layout="C", queue=queue_ty)
+    kernel_sig = types.void(i64arr_ty)
+
+    disp = exp_dpex.kernel(pass_flags_to_func)
+    disp.compile(kernel_sig)
+    kcres = disp.overloads[kernel_sig.args]
+    llvm_ir_mod = kcres.library._final_module.__str__()
+
+    pattern = re.compile(
+        r"call spir_func i32 @\_Z.*bitwise\_or"
+        r"\_flags.*\(i64\* nonnull %.*, i64 1, i64 2\)"
+    )
+
+    assert re.search(pattern, llvm_ir_mod) is not None