import itertools import numpy as np import unittest from numba.core.compiler import compile_isolated, Flags from numba.core import types, utils from numba.tests import usecases from numba.tests.support import TestCase, tag enable_pyobj_flags = Flags() enable_pyobj_flags.enable_pyobject = True force_pyobj_flags = Flags() force_pyobj_flags.force_pyobject = True class TestUsecases(TestCase): def test_andor(self): pyfunc = usecases.andor cr = compile_isolated(pyfunc, (types.int32, types.int32)) cfunc = cr.entry_point # Argument boundaries xs = -1, 0, 1, 9, 10, 11 ys = -1, 0, 1, 9, 10, 11 for args in itertools.product(xs, ys): self.assertEqual(pyfunc(*args), cfunc(*args), "args %s" % (args,)) def test_sum1d(self): pyfunc = usecases.sum1d cr = compile_isolated(pyfunc, (types.int32, types.int32)) cfunc = cr.entry_point ss = -1, 0, 1, 100, 200 es = -1, 0, 1, 100, 200 for args in itertools.product(ss, es): self.assertEqual(pyfunc(*args), cfunc(*args), args) def test_sum1d_pyobj(self): pyfunc = usecases.sum1d cr = compile_isolated(pyfunc, (types.int32, types.int32), flags=force_pyobj_flags) cfunc = cr.entry_point ss = -1, 0, 1, 100, 200 es = -1, 0, 1, 100, 200 for args in itertools.product(ss, es): self.assertEqual(pyfunc(*args), cfunc(*args), args) args = 0, 500 def bm_python(): pyfunc(*args) def bm_numba(): cfunc(*args) print(utils.benchmark(bm_python, maxsec=.1)) print(utils.benchmark(bm_numba, maxsec=.1)) def test_sum2d(self): pyfunc = usecases.sum2d cr = compile_isolated(pyfunc, (types.int32, types.int32)) cfunc = cr.entry_point ss = -1, 0, 1, 100, 200 es = -1, 0, 1, 100, 200 for args in itertools.product(ss, es): self.assertEqual(pyfunc(*args), cfunc(*args), args) def test_while_count(self): pyfunc = usecases.while_count cr = compile_isolated(pyfunc, (types.int32, types.int32)) cfunc = cr.entry_point ss = -1, 0, 1, 100, 200 es = -1, 0, 1, 100, 200 for args in itertools.product(ss, es): self.assertEqual(pyfunc(*args), cfunc(*args), args) def test_copy_arrays(self): pyfunc = usecases.copy_arrays arraytype = types.Array(types.int32, 1, 'A') cr = compile_isolated(pyfunc, (arraytype, arraytype)) cfunc = cr.entry_point nda = 0, 1, 10, 100 for nd in nda: a = np.arange(nd, dtype='int32') b = np.empty_like(a) args = a, b cfunc(*args) self.assertPreciseEqual(a, b, msg=str(args)) def test_copy_arrays2d(self): pyfunc = usecases.copy_arrays2d arraytype = types.Array(types.int32, 2, 'A') cr = compile_isolated(pyfunc, (arraytype, arraytype)) cfunc = cr.entry_point nda = (0, 0), (1, 1), (2, 5), (4, 25) for nd in nda: d1, d2 = nd a = np.arange(d1 * d2, dtype='int32').reshape(d1, d2) b = np.empty_like(a) args = a, b cfunc(*args) self.assertPreciseEqual(a, b, msg=str(args)) def run_ifelse(self, pyfunc): cr = compile_isolated(pyfunc, (types.int32, types.int32)) cfunc = cr.entry_point xs = -1, 0, 1 ys = -1, 0, 1 for x, y in itertools.product(xs, ys): args = x, y self.assertEqual(pyfunc(*args), cfunc(*args), args) def test_string_concat(self): pyfunc = usecases.string_concat cr = compile_isolated(pyfunc, (types.int32, types.int32), flags=enable_pyobj_flags) cfunc = cr.entry_point xs = -1, 0, 1 ys = -1, 0, 1 for x, y in itertools.product(xs, ys): args = x, y self.assertEqual(pyfunc(*args), cfunc(*args), args) def test_string_len(self): pyfunc = usecases.string_len cr = compile_isolated(pyfunc, (types.pyobject,), flags=enable_pyobj_flags) cfunc = cr.entry_point test_str = '123456' self.assertEqual(pyfunc(test_str), cfunc(test_str)) test_str = '1' self.assertEqual(pyfunc(test_str), cfunc(test_str)) test_str = '' self.assertEqual(pyfunc(test_str), cfunc(test_str)) def test_string_slicing(self): pyfunc = usecases.string_slicing cr = compile_isolated(pyfunc, (types.pyobject,) * 3, flags=enable_pyobj_flags) cfunc = cr.entry_point test_str = '123456' self.assertEqual(pyfunc(test_str, 0, 3), cfunc(test_str, 0, 3)) self.assertEqual(pyfunc(test_str, 1, 5), cfunc(test_str, 1, 5)) self.assertEqual(pyfunc(test_str, 2, 3), cfunc(test_str, 2, 3)) def test_string_conversion(self): pyfunc = usecases.string_conversion cr = compile_isolated(pyfunc, (types.int32,), flags=enable_pyobj_flags) cfunc = cr.entry_point self.assertEqual(pyfunc(1), cfunc(1)) cr = compile_isolated(pyfunc, (types.float32,), flags=enable_pyobj_flags) cfunc = cr.entry_point self.assertEqual(pyfunc(1.1), cfunc(1.1)) def test_string_comparisons(self): import operator pyfunc = usecases.string_comparison cr = compile_isolated(pyfunc, (types.pyobject, types.pyobject, types.pyobject), flags=enable_pyobj_flags) cfunc = cr.entry_point test_str1 = '123' test_str2 = '123' op = operator.eq self.assertEqual(pyfunc(test_str1, test_str2, op), cfunc(test_str1, test_str2, op)) test_str1 = '123' test_str2 = '456' op = operator.eq self.assertEqual(pyfunc(test_str1, test_str2, op), cfunc(test_str1, test_str2, op)) test_str1 = '123' test_str2 = '123' op = operator.ne self.assertEqual(pyfunc(test_str1, test_str2, op), cfunc(test_str1, test_str2, op)) test_str1 = '123' test_str2 = '456' op = operator.ne self.assertEqual(pyfunc(test_str1, test_str2, op), cfunc(test_str1, test_str2, op)) def test_blackscholes_cnd(self): pyfunc = usecases.blackscholes_cnd cr = compile_isolated(pyfunc, (types.float32,)) cfunc = cr.entry_point ds = -0.5, 0, 0.5 for d in ds: args = (d,) self.assertEqual(pyfunc(*args), cfunc(*args), args) if __name__ == '__main__': unittest.main()