Merge 9e189e6 into 5b91644

numba_pandas/__init__.py

@@ -0,0 +1,11 @@
+
+def _init_extension():
+    '''Register Pandas classes and functions with Numba.
+
+    This exntry_point is called by Numba when it initializes.
+    '''
+    # TODO: import all packages overloading Pandas data types and functions
+    import hpat.hiframes.api
+    import hpat.hiframes.boxing
+    # imported from above packages
+    # import hpat.hiframes.pd_series_ext

numba_pandas/tests/test_pandas.py

@@ -0,0 +1,24 @@
+import numba
+import pandas
+
+from packaging import version
+
+# numba <0.46 requires explicit import extension
+if version.parse(numba.__version__) < version.parse('0.46'):
+    from numba_pandas import _init_extension
+    _init_extension()
+
+
+def test_function():
+    pandas_func = pandas.Series
+
+    @numba.njit
+    def numba_func(*args):
+        return pandas_func(*args)
+
+    args = [0, 1]
+
+    overload_value = numba_func(args)
+    pandas_value = pandas_func(args)
+
+    pandas.testing.assert_series_equal(overload_value, pandas_value)

setup.py

@@ -528,4 +528,10 @@ def run(self):
       install_requires=['numba'],
       extras_require={'HDF5': ["h5py"], 'Parquet': ["pyarrow"]},
       cmdclass=hpat_build_commands,
-      ext_modules=_ext_mods)
+      ext_modules=_ext_mods,
+      entry_points={
+          "numba_extensions": [
+              "init = numba_pandas:_init_extension",
+          ],
+      },
+)