ai-content-maker/.venv/Lib/site-packages/numba/tests/test_mangling.py

42 lines
1.2 KiB
Python

# -*- 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()