Series copy numpy-like call (#588)

* numpy-like copy

* Fix dtype initial

* fix impl bool+number

* small fix

* fix tests

* Replace copy call

* pep fix

Author: 1e-to

sdc/datatypes/hpat_pandas_series_functions.py

@@ -2129,17 +2129,18 @@ def hpat_pandas_series_copy(self, deep=True):
     if isinstance(self.index, types.NoneType):
         def hpat_pandas_series_copy_impl(self, deep=True):
             if deep:
-                return pandas.Series(data=self._data.copy(), name=self._name)
+                return pandas.Series(data=numpy_like.copy(self._data), name=self._name)
             else:
                 return pandas.Series(data=self._data, name=self._name)
         return hpat_pandas_series_copy_impl
     else:
         def hpat_pandas_series_copy_impl(self, deep=True):
             if deep:
-                return pandas.Series(data=self._data.copy(), index=self._index.copy(), name=self._name)
+                return pandas.Series(data=numpy_like.copy(self._data), index=numpy_like.copy(self._index),
+                                     name=self._name)
             else:
                 # Shallow copy of index is not supported yet
-                return pandas.Series(data=self._data, index=self._index.copy(), name=self._name)
+                return pandas.Series(data=self._data, index=numpy_like.copy(self._index), name=self._name)
         return hpat_pandas_series_copy_impl
 
 

sdc/functions/numpy_like.py

@@ -140,7 +140,7 @@ def sdc_copy_overload(self):
        Test: python -m sdc.runtests sdc.tests.test_sdc_numpy -k copy
     """
 
-    if not isinstance(self, types.Array):
+    if not isinstance(self, (types.Array, StringArrayType)):
         return None
 
     dtype = self.dtype
@@ -155,7 +155,7 @@ def sdc_copy_number_impl(self):
 
         return sdc_copy_number_impl
 
-    if isinstance(dtype, types.npytypes.UnicodeCharSeq):
+    if isinstance(dtype, (types.npytypes.UnicodeCharSeq, types.UnicodeType, types.StringLiteral)):
         def sdc_copy_string_impl(self):
             return self.copy()
 

sdc/tests/tests_perf/test_perf_series.py

@@ -73,7 +73,7 @@ def _test_case(self, pyfunc, name, total_data_length, data_num=1, input_data=tes
     TC(name='at', size=[10 ** 7], call_expr='data.at[3]', usecase_params='data'),
     TC(name='chain_add_and_sum', size=[20 * 10 ** 6, 25 * 10 ** 6, 30 * 10 ** 6], call_expr='(A + B).sum()',
        usecase_params='A, B', data_num=2),
-    TC(name='copy', size=[10 ** 8]),
+    TC(name='copy', size=[10 ** 7]),
     TC(name='corr',  size=[10 ** 7],params='other', data_num=2),
     TC(name='count', size=[10 ** 8]),
     TC(name='cov', size=[10 ** 8], params='other', data_num=2),