Viewing File: /home/ubuntu/combine_ai/combine/lib/python3.10/site-packages/numba/tests/test_locals.py

from numba import float32, njit
import unittest


def foo():
    x = 123
    return x


class TestLocals(unittest.TestCase):

    def test_seed_types(self):
        cfunc = njit((), locals={'x': float32})(foo)
        self.assertEqual(cfunc.nopython_signatures[0].return_type, float32)


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