More tests

Author: sfc-gh-jkew

modin/core/storage_formats/pandas/query_compiler_caster.py

@@ -64,7 +64,9 @@ def calculate(self):
             costs_2 = qc_2.qc_engine_switch_cost(qc_1)
             self._add_cost_data(costs_1)
             self._add_cost_data(costs_2)
-        
+        if len(self._caster_costing_map) <= 0 and len(self._qc_cls_list) > 0:
+            self._result_type = self._qc_cls_list[0]
+            return self._result_type
         min_value = min(self._caster_costing_map.values())
         for key, value in self._caster_costing_map.items():
             if min_value == value:
@@ -163,16 +165,16 @@ def apply_argument_cast(obj: Fn) -> Fn:
     """
     if isinstance(obj, type):
         all_attrs = dict(inspect.getmembers(obj))
-        all_attrs.pop("__abstractmethods__")
-        all_attrs.pop("__init__")
-        all_attrs.pop("qc_engine_switch_cost")
-        all_attrs.pop("from_pandas")
+
 
         # This is required because inspect converts class methods to member functions
         current_class_attrs = vars(obj)
         for key in current_class_attrs:
             all_attrs[key] = current_class_attrs[key]
-
+        all_attrs.pop("__abstractmethods__")
+        all_attrs.pop("__init__")
+        all_attrs.pop("qc_engine_switch_cost")
+        all_attrs.pop("from_pandas")
         for attr_name, attr_value in all_attrs.items():
             if isinstance(
                 attr_value, (FunctionType, MethodType, classmethod, staticmethod)
@@ -203,7 +205,6 @@ def cast_args(*args: Tuple, **kwargs: Dict) -> Any:
         """
         if len(args) == 0 and len(kwargs) == 0:
             return
-        print(f"Adding wrapper {obj}\n")
         current_qc = args[0]
         calculator = QueryCompilerCasterCalculator()
         calculator.add_query_compiler(current_qc)

modin/tests/pandas/native_df_interoperability/test_compiler_caster.py

@@ -74,10 +74,18 @@ def test_two_two_qc_types_lhs():
     assert(type(df3) == type(df2)) # should move to cluster
 
 def test_three_two_qc_types_rhs():
-    pass
+    df = CloudQC(pandas.DataFrame([0, 1, 2]))
+    df2 = CloudQC(pandas.DataFrame([0, 1, 2]))
+    df3 = PicoQC(pandas.DataFrame([0, 1, 2]))
+    df4 = df3.concat(axis=1, other=[df, df2])
+    assert(type(df) == type(df4)) # should move to cloud
 
 def test_three_two_qc_types_lhs():
-    pass
+    df = CloudQC(pandas.DataFrame([0, 1, 2]))
+    df2 = CloudQC(pandas.DataFrame([0, 1, 2]))
+    df3 = PicoQC(pandas.DataFrame([0, 1, 2]))
+    df4 = df.concat(axis=1, other=[df2, df3])
+    assert(type(df) == type(df4)) # should move to cloud 
 
 def test_three_two_qc_types_middle():
     pass