Videre

linedance-app/venv/lib/python3.12/site-packages/numba/tests/test_repr.py

@@ -0,0 +1,63 @@
+import unittest
+import numpy as np
+
+from numba.tests.support import TestCase
+from numba import typeof
+from numba.core import types
+from numba.typed import List, Dict
+
+NB_TYPES = [
+    types.Array,
+    types.NestedArray,
+    types.bool_,
+    types.unicode_type,
+    types.Record,
+    types.UnicodeCharSeq,
+    types.UniTuple,
+    types.List,
+    types.Tuple,
+    types.DictType,
+    types.ListType,
+    types.Set,
+] + list(types.number_domain)
+
+
+class TestRepr(TestCase):
+    def setUp(self) -> None:
+        tys_ns = {ty.__name__: ty for ty in NB_TYPES if hasattr(ty, "__name__")}
+        tys_ns.update({ty.name: ty for ty in NB_TYPES if hasattr(ty, "name")})
+        self.tys_ns = tys_ns
+
+    def check_repr(self, val):
+        ty = typeof(val)
+        ty2 = eval(repr(ty), self.tys_ns)
+        self.assertEqual(ty, ty2)
+
+    def test_types(self):
+        # define some values for the test cases
+        rec_dtype = [("a", "f8"), ("b", "U8"), ("c", "i8", (2, 3))]
+        nb_dict = Dict()
+        nb_dict['a'] = 1
+        # tests cases: list of different types + list comp of number types
+        val_types_cases = [
+            True,
+            "a",
+            (1, 2),
+            (1, "a"),
+            [1, "a"],
+            ([1, "a"], [2, "b"]),
+            ((1, 2), (3, "b")),
+            ((1, 2), (3, [1, 2])),
+            np.ones(3),
+            np.array([(1, "a", np.ones((2, 3)))], dtype=rec_dtype),
+            nb_dict,
+            List([1, 2]),
+            {1, 2},
+        ] + [number(1.1) for number in types.number_domain]
+
+        for val in val_types_cases:
+            self.check_repr(val)
+
+
+if __name__ == '__main__':
+    unittest.main()