42 lines
1.2 KiB
Python
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()
|