Unit tests

Author: Diptorup Deb

numba_dpex/tests/experimental/IntEnumLiteral/__init__.py

@@ -0,0 +1,30 @@
+# SPDX-FileCopyrightText: 2023 Intel Corporation
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from enum import IntEnum
+
+import pytest
+
+from numba_dpex.core.exceptions import IllegalIntEnumLiteralValueError
+from numba_dpex.experimental import IntEnumLiteral
+from numba_dpex.experimental.flag_enum import FlagEnum
+
+
+def test_intenumliteral_creation():
+    """Tests the creation of an IntEnumLiteral type."""
+
+    class DummyFlags(FlagEnum):
+        DUMMY = 0
+
+    try:
+        IntEnumLiteral(DummyFlags)
+    except:
+        pytest.fail("Unexpected failure in IntEnumLiteral initialization")
+
+    with pytest.raises(IllegalIntEnumLiteralValueError):
+
+        class SomeKindOfUnknownEnum(IntEnum):
+            UNKNOWN_FLAG = 1
+
+        IntEnumLiteral(SomeKindOfUnknownEnum)

numba_dpex/tests/experimental/IntEnumLiteral/test_type_creation.py

@@ -0,0 +1,36 @@
+# SPDX-FileCopyrightText: 2023 Intel Corporation
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import pytest
+from numba.core.datamodel import default_manager
+
+from numba_dpex.core.datamodel.models import dpex_data_model_manager
+from numba_dpex.experimental import IntEnumLiteral
+from numba_dpex.experimental.flag_enum import FlagEnum
+from numba_dpex.experimental.models import exp_dmm
+
+
+def test_data_model_registration():
+    """Tests that the IntEnumLiteral type is only registered with the
+    DpexExpKernelTargetContext target.
+    """
+
+    class DummyFlags(FlagEnum):
+        DUMMY = 0
+
+    dummy = IntEnumLiteral(DummyFlags)
+
+    with pytest.raises(KeyError):
+        default_manager.lookup(dummy)
+
+    with pytest.raises(KeyError):
+        dpex_data_model_manager.lookup(dummy)
+
+    try:
+        exp_dmm.lookup(dummy)
+    except:
+        pytest.fail(
+            "IntEnumLiteral type lookup failed in experimental "
+            "data model manager"
+        )