bool for series isna (#586)

Author: 1e-to

sdc/datatypes/hpat_pandas_series_functions.py

@@ -2395,7 +2395,7 @@ def hpat_pandas_series_isnull(self):
     ty_checker = TypeChecker(_func_name)
     ty_checker.check(self, SeriesType)
 
-    if isinstance(self.data.dtype, (types.Integer, types.Float)):
+    if isinstance(self.data.dtype, (types.Number, types.Boolean, bool)):
         def hpat_pandas_series_isnull_impl(self):
             return pandas.Series(data=numpy.isnan(self._data), index=self._index, name=self._name)
 
@@ -2463,7 +2463,7 @@ def hpat_pandas_series_isna(self):
     ty_checker = TypeChecker(_func_name)
     ty_checker.check(self, SeriesType)
 
-    if isinstance(self.data.dtype, (types.Integer, types.Float)):
+    if isinstance(self.data.dtype, (types.Number, types.Boolean, bool)):
         def hpat_pandas_series_isna_impl(self):
             return pandas.Series(data=numpy.isnan(self._data), index=self._index, name=self._name)
 
@@ -2531,7 +2531,7 @@ def hpat_pandas_series_notna(self):
     ty_checker = TypeChecker(_func_name)
     ty_checker.check(self, SeriesType)
 
-    if isinstance(self.data.dtype, types.Number):
+    if isinstance(self.data.dtype, (types.Number, types.Boolean, bool)):
         def hpat_pandas_series_notna_impl(self):
             return pandas.Series(numpy.invert(numpy.isnan(self._data)), index=self._index, name=self._name)
 

sdc/functions/numpy_like.py

@@ -163,7 +163,7 @@ def sdc_copy_string_impl(self):
 
 
 @sdc_overload(notnan)
-def sdc_isnan_overload(self):
+def sdc_notnan_overload(self):
     """
     Intel Scalable Dataframe Compiler Developer Guide
     *************************************************
@@ -177,7 +177,7 @@ def sdc_isnan_overload(self):
 
     dtype = self.dtype
     isnan = get_isnan(dtype)
-    if isinstance(dtype, types.Integer):
+    if isinstance(dtype, (types.Integer, types.Boolean, bool)):
         def sdc_notnan_int_impl(self):
             length = len(self)
             res = numpy.ones(shape=length, dtype=numpy.bool_)
@@ -215,7 +215,7 @@ def sdc_isnan_overload(self):
 
     dtype = self.dtype
     isnan = get_isnan(dtype)
-    if isinstance(dtype, types.Integer):
+    if isinstance(dtype, (types.Integer, types.Boolean, bool)):
         def sdc_isnan_int_impl(self):
             length = len(self)
             res = numpy.zeros(shape=length, dtype=numpy.bool_)

sdc/tests/test_series.py

@@ -3582,7 +3582,7 @@ def test_impl(S):
 
         jit_func = self.jit(test_impl)
 
-        datas = [[0, 1, 2, 3], [0., 1., np.inf, np.nan], ['a', None, 'b', 'c']]
+        datas = [[0, 1, 2, 3], [0., 1., np.inf, np.nan], ['a', None, 'b', 'c'], [True, True, False, False]]
         indices = [None, [3, 2, 1, 0], ['a', 'b', 'c', 'd']]
         names = [None, 'A']
 
@@ -3600,7 +3600,7 @@ def test_impl(S):
 
         jit_func = self.jit(test_impl)
 
-        datas = [[0, 1, 2, 3], [0., 1., np.inf, np.nan], ['a', None, 'b', 'c']]
+        datas = [[0, 1, 2, 3], [0., 1., np.inf, np.nan], ['a', None, 'b', 'c'], [True, True, False, False]]
         indices = [None, [3, 2, 1, 0], ['a', 'b', 'c', 'd']]
         names = [None, 'A']
 
@@ -3626,7 +3626,7 @@ def test_impl(S):
 
         jit_func = self.jit(test_impl)
 
-        datas = [[0, 1, 2, 3], [0., 1., np.inf, np.nan], ['a', None, 'b', 'c']]
+        datas = [[0, 1, 2, 3], [0., 1., np.inf, np.nan], ['a', None, 'b', 'c'], [True, True, False, False]]
         indices = [None, [3, 2, 1, 0], ['a', 'b', 'c', 'd']]
         names = [None, 'A']