test_mangling.py

# -*- 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.assertRegexpMatches(name, r'^_Z[a-zA-Z0-9_\$]+$')


if __name__ == '__main__':
    unittest.main()