Minor change in 1 test on dataframe

densmirn · densmirn · commit e663d33cffe6 · 2020-05-27T12:26:54.000+03:00
diff --git a/sdc/hiframes/boxing.py b/sdc/hiframes/boxing.py
@@ -247,7 +247,8 @@ def box_dataframe(typ, val, c):
             # context.nrt.incref(builder, arr_typ, arr)
         pyapi.dict_setitem(df_dict, cname_obj, arr_obj)
 
-        c.context.nrt.decref(c.builder, list_type, typ_arrs.value)
+        if c.context.enable_nrt:
+            c.context.nrt.decref(c.builder, list_type, list_val)
         pyapi.decref(arr_obj)
         pyapi.decref(cname_obj)
 
diff --git a/sdc/tests/test_dataframe.py b/sdc/tests/test_dataframe.py
@@ -2333,15 +2333,19 @@ def test_append_df_diff_cols_index_ignore_false_no_unboxing(self):
         def test_impl():
             n1 = 11
             n2 = n1 * 2
-            df = pd.DataFrame({'A': np.arange(n1), 'B': np.arange(n1) ** 2},
-                              index=np.arange(n1) ** 2)
-            df2 = pd.DataFrame({'C': np.arange(n2), 'D': np.arange(n2) ** 2,
-                                'E S D': np.arange(n2) + 100},
-                               index=np.arange(n2) ** 4)
+            df = pd.DataFrame({
+                'A': np.arange(n1), 'B': np.arange(n1) ** 2
+            }, index=np.arange(n1) ** 2)
+            df2 = pd.DataFrame({
+                'C': np.arange(n2), 'D': np.arange(n2) ** 2,
+                'E S D': np.arange(n2) + 100
+            }, index=np.arange(n2) ** 4)
             return df.append(df2, ignore_index=False)
 
         sdc_func = self.jit(test_impl)
-        pd.testing.assert_frame_equal(sdc_func(), test_impl())
+        res_jit = sdc_func()
+        res_ref = test_impl()
+        pd.testing.assert_frame_equal(res_jit, res_ref)
 
     @dfRefactoringNotImplemented  # required re-implementing DataFrame unboxing
     def test_append_df_diff_cols_index_ignore_index(self):
