Fix dt_day_of_week/day_of_year, str_cat/extract/partition/replace/rpa…

…rtition (modin-project#51)

* Fix dt_day_of_week/day_of_year, str_partition/replace/rpartition

* Fix str_extract

modin/core/execution/client/query_compiler.py

@@ -659,9 +659,11 @@ def forwarding_method(self, by, *args, **kwargs):
         "dt_minute",
         "dt_hour",
         "dt_day",
+        "dt_day_of_week",
         "dt_dayofweek",
         "dt_weekday",
         "dt_day_name",
+        "dt_day_of_year",
         "dt_dayofyear",
         "dt_week",
         "dt_weekofyear",

modin/pandas/series_utils.py

@@ -141,8 +141,14 @@ def casefold(self):
     def cat(self, others=None, sep=None, na_rep=None, join=None):
         if isinstance(others, Series):
             others = others._to_pandas()
-        return self._default_to_pandas(
-            pandas.Series.str.cat, others=others, sep=sep, na_rep=na_rep, join=join
+        data = Series(query_compiler=self._query_compiler)
+        return data._reduce_dimension(
+            self._query_compiler.str_cat(
+                others=others,
+                sep=sep,
+                na_rep=na_rep,
+                join=join
+            )
         )
 
     def decode(self, encoding, errors="strict"):
@@ -307,6 +313,13 @@ def match(self, pat, case=True, flags=0, na=np.NaN):
         )
 
     def extract(self, pat, flags=0, expand=True):
+        import re
+        n = re.compile(pat).groups
+        if expand or n > 1:
+            from .dataframe import DataFrame
+            return DataFrame(
+                query_compiler=self._query_compiler.str_extract(pat, flags, expand)
+            )
         return Series(
             query_compiler=self._query_compiler.str_extract(pat, flags, expand)
         )
@@ -329,9 +342,20 @@ def lstrip(self, to_strip=None):
     def partition(self, sep=" ", expand=True):
         if sep is not None and len(sep) == 0:
             raise ValueError("empty separator")
-        return Series(
-            query_compiler=self._query_compiler.str_partition(sep=sep, expand=expand)
-        )
+
+        if expand:
+            from .dataframe import DataFrame
+            return DataFrame(
+                query_compiler=self._query_compiler.str_partition(
+                    sep=sep, expand=expand
+                )
+            )
+        else:
+            return Series(
+                query_compiler=self._query_compiler.str_partition(
+                    sep=sep, expand=expand
+                )
+            )
 
     def removeprefix(self, prefix):
         return Series(query_compiler=self._query_compiler.str_removeprefix(prefix))
@@ -343,11 +367,22 @@ def repeat(self, repeats):
         return Series(query_compiler=self._query_compiler.str_repeat(repeats))
 
     def rpartition(self, sep=" ", expand=True):
-        return Series(
-            query_compiler=self._query_compiler.str_rpartition(
-                sep=sep, expand=expand
+        if sep is not None and len(sep) == 0:
+            raise ValueError("empty separator")
+
+        if expand:
+            from .dataframe import DataFrame
+            return DataFrame(
+                query_compiler=self._query_compiler.str_rpartition(
+                    sep=sep, expand=expand
+                )
+            )
+        else:
+            return Series(
+                query_compiler=self._query_compiler.str_rpartition(
+                    sep=sep, expand=expand
+                )
             )
-        )
 
     def lower(self):
         return Series(query_compiler=self._query_compiler.str_lower())