Refactor function handling if checks in distributed_analysis.py part 1 (

#135)
bodo-ai · Jan 13, 2025 · 9974ea1 · 9974ea1
1 parent 6f19f08
commit 9974ea1
Show file tree

Hide file tree

Showing 3 changed files with 1,502 additions and 970 deletions.
diff --git a/bodo/tests/test_dataframe_part1.py b/bodo/tests/test_dataframe_part1.py
@@ -1276,7 +1276,7 @@ def test_df_rename_all_types(df_value, memory_leak_check):
     """
 
     def test_impl(df):
-        df.rename(columns={"A": "first", "B": "second"})
+        return df.rename(columns={"A": "first", "B": "second"})
 
     check_func(test_impl, (df_value,))
 
@@ -1289,7 +1289,7 @@ def test_df_rename_mapper_all_types(df_value, memory_leak_check):
     """
 
     def test_impl(df):
-        df.rename({"A": "first", "B": "second"}, axis=1)
+        return df.rename({"A": "first", "B": "second"}, axis=1)
 
     check_func(test_impl, (df_value,))