SNOW-2430578: Add support for invert in faster pandas (#3895)

Author: sfc-gh-helmeleegy

CHANGELOG.md

@@ -112,6 +112,7 @@
   - `loc` (setting columns)
   - `to_datetime`
   - `drop`
+  - `invert`
 - Reuse row count from the relaxed query compiler in `get_axis_len`.
 
 #### Bug Fixes

src/snowflake/snowpark/modin/plugin/compiler/snowflake_query_compiler.py

@@ -11434,6 +11434,16 @@ def transpose(self) -> "SnowflakeQueryCompiler":
         return SnowflakeQueryCompiler(new_internal_frame)
 
     def invert(self) -> "SnowflakeQueryCompiler":
+        """
+        Wrapper around _invert_internal to be supported in faster pandas.
+        """
+        relaxed_query_compiler = None
+        if self._relaxed_query_compiler is not None:
+            relaxed_query_compiler = self._relaxed_query_compiler._invert_internal()
+        qc = self._invert_internal()
+        return self._maybe_set_relaxed_qc(qc, relaxed_query_compiler)
+
+    def _invert_internal(self) -> "SnowflakeQueryCompiler":
         """
         Apply bitwise inversion for each element of the QueryCompiler.
 

tests/integ/modin/test_faster_pandas.py

@@ -223,6 +223,35 @@ def test_drop(session):
     assert_frame_equal(snow_result, native_result)
 
 
+@sql_count_checker(query_count=3)
+def test_invert(session):
+    # create tables
+    table_name = Utils.random_name_for_temp_object(TempObjectType.TABLE)
+    session.create_dataframe(
+        native_pd.DataFrame([[2, True], [1, False], [3, False]], columns=["A", "B"])
+    ).write.save_as_table(table_name, table_type="temp")
+
+    # create snow dataframes
+    df = pd.read_snowflake(table_name)
+    snow_result = ~df["B"]
+
+    # verify that the input dataframe has a populated relaxed query compiler
+    assert df._query_compiler._relaxed_query_compiler is not None
+    assert df._query_compiler._relaxed_query_compiler._dummy_row_pos_mode is True
+    # verify that the output dataframe also has a populated relaxed query compiler
+    assert snow_result._query_compiler._relaxed_query_compiler is not None
+    assert (
+        snow_result._query_compiler._relaxed_query_compiler._dummy_row_pos_mode is True
+    )
+
+    # create pandas dataframes
+    native_df = df.to_pandas()
+    native_result = ~native_df["B"]
+
+    # compare results
+    assert_series_equal(snow_result, native_result)
+
+
 @pytest.mark.parametrize("func", ["isna", "isnull", "notna", "notnull"])
 @sql_count_checker(query_count=3)
 def test_isna_notna(session, func):