Videre

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

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+"""
+Test function name mangling.
+The mangling affects the ABI of numba compiled binaries.
+"""
+from numba.core import types
+from numba.core.funcdesc import default_mangler
+from numba.tests.support import unittest, TestCase
+
+
+class TestMangling(TestCase):
+    def test_one_args(self):
+        fname = 'foo'
+        argtypes = types.int32,
+        name = default_mangler(fname, argtypes)
+        self.assertEqual(name, '_Z3fooi')
+
+    def test_two_args(self):
+        fname = 'foo'
+        argtypes = types.int32, types.float32
+        name = default_mangler(fname, argtypes)
+        self.assertEqual(name, '_Z3fooif')
+
+    def test_unicode_fname(self):
+        fname = u'foಠ'
+        argtypes = types.int32, types.float32
+        name = default_mangler(fname, argtypes)
+        self.assertIsInstance(name, str)
+        # manually encode it
+        unichar = fname[2]
+        enc = ''.join('_{:02x}'.format(c)
+                      for c in unichar.encode('utf8'))
+        text = 'fo' + enc
+        expect = '_Z{}{}if'.format(len(text), text)
+        self.assertEqual(name, expect)
+        # ensure result chars are in the right charset
+        self.assertRegex(name, r'^_Z[a-zA-Z0-9_\$]+$')
+
+
+if __name__ == '__main__':
+    unittest.main()