Add additional backends

Author: karangattu

pyproject.toml

@@ -87,6 +87,10 @@ test = [
     "faicons",
     "ridgeplot",
     "great_tables",
+    "modin",
+    "polars",
+    "dask[dataframe]",
+    "pyarrow",
 ]
 dev = [
     "black>=24.0",

tests/playwright/shiny/components/data_frame/data_type/app.py

@@ -1,18 +1,32 @@
 from __future__ import annotations
 
+import dask.dataframe as dd
+import modin.pandas as mpd  # pyright: ignore[reportMissingTypeStubs]
 import narwhals.stable.v1 as nw
 import palmerpenguins  # pyright: ignore[reportMissingTypeStubs]
+import polars as pl
+import pyarrow as pa
 
 from shiny.express import render, ui
 
+# import databricks.koalas as ks  # pyright: ignore[reportMissingTypeStubs]
+
+
 pd_df = palmerpenguins.load_penguins_raw().iloc[0:2, 0:2]
 
 nw_df = nw.from_native(pd_df, eager_only=True)
+pa_df = pa.Table.from_pandas(pd_df)
+mpd_df = mpd.DataFrame(pd_df)
+pl_df = pl.DataFrame(pd_df)
+dd_df = dd.from_pandas(
+    pd_df
+)  # pyright: ignore[reportUnknownMemberType, reportPrivateImportUsage, reportUnknownVariableType]
+# ks_df = ks.from_pandas(pd_df)  # pyright: ignore[reportMissingTypeStubs, reportUnknownMemberType, reportUnknownVariableType]
+
 
 with ui.layout_columns():
 
     with ui.card():
-
         ui.h2("Original Pandas Data")
 
         @render.data_frame
@@ -73,3 +87,158 @@ def nw_data_view():
         @render.code
         def nw_data_view_selected():
             return str(type(nw_df_original.data_view(selected=True)))
+
+    with ui.card():
+        ui.h2("Original PyArrow Data")
+
+        @render.data_frame
+        def pa_df_original():
+            return pa_df
+
+        "Data type:"
+
+        @render.code
+        def pa_type():
+            return str(type(pa_df))
+
+        ui.markdown("`.data()` type:")
+
+        @render.code
+        def pa_data():
+            return str(type(pa_df_original.data()))
+
+        ui.markdown("`.data_view()` type:")
+
+        @render.code
+        def pa_data_view():
+            return str(type(pa_df_original.data_view()))
+
+        ui.markdown("`.data_view(selected=True)` type:")
+
+        @render.code
+        def pa_data_view_selected():
+            return str(type(pa_df_original.data_view(selected=True)))
+
+    with ui.card():
+        ui.h2("Modin Data")
+
+        @render.data_frame
+        def mpd_df_original():
+            return mpd_df
+
+        "Data type:"
+
+        @render.code
+        def mpd_type():
+            return str(type(mpd_df))
+
+        ui.markdown("`.data()` type:")
+
+        @render.code
+        def mpd_data():
+            return str(type(mpd_df_original.data()))
+
+        ui.markdown("`.data_view()` type:")
+
+        @render.code
+        def mpd_data_view():
+            return str(type(mpd_df_original.data_view()))
+
+        ui.markdown("`.data_view(selected=True)` type:")
+
+        @render.code
+        def mpd_data_view_selected():
+            return str(type(mpd_df_original.data_view(selected=True)))
+
+    with ui.card():
+        ui.h2("Polars Data")
+
+        @render.data_frame
+        def pl_df_original():
+            return pl_df
+
+        "Data type:"
+
+        @render.code
+        def pl_type():
+            return str(type(pl_df))
+
+        ui.markdown("`.data()` type:")
+
+        @render.code
+        def pl_data():
+            return str(type(pl_df_original.data()))
+
+        ui.markdown("`.data_view()` type:")
+
+        @render.code
+        def pl_data_view():
+            return str(type(pl_df_original.data_view()))
+
+        ui.markdown("`.data_view(selected=True)` type:")
+
+        @render.code
+        def pl_data_view_selected():
+            return str(type(pl_df_original.data_view(selected=True)))
+
+    with ui.card():
+        ui.h2("Dask Data")
+
+        @render.data_frame
+        def dd_df_original():
+            return dd_df
+
+        "Data type:"
+
+        @render.code
+        def dd_type():
+            return str(type(dd_df))
+
+        ui.markdown("`.data()` type:")
+
+        @render.code
+        def dd_data():
+            return str(type(dd_df_original.data()))
+
+        ui.markdown("`.data_view()` type:")
+
+        @render.code
+        def dd_data_view():
+            return str(type(dd_df_original.data_view()))
+
+        ui.markdown("`.data_view(selected=True)` type:")
+
+        @render.code
+        def dd_data_view_selected():
+            return str(type(dd_df_original.data_view(selected=True)))
+
+    # with ui.card():
+    #     ui.h2("Koalas Data")
+
+    #     @render.data_frame
+    #     def ks_df_original():
+    #         return ks_df
+
+    #     "Data type:"
+
+    #     @render.code
+    #     def ks_type():
+    #         return str(type(ks_df))
+
+    #     ui.markdown("`.data()` type:")
+
+    #     @render.code
+    #     def ks_data():
+    #         return str(type(ks_df_original.data()))
+
+    #     ui.markdown("`.data_view()` type:")
+
+    #     @render.code
+    #     def ks_data_view():
+    #         return str(type(ks_df_original.data_view()))
+
+    #     ui.markdown("`.data_view(selected=True)` type:")
+
+    #     @render.code
+    #     def ks_data_view_selected():
+    #         return str(type(ks_df_original.data_view(selected=True)))

tests/playwright/shiny/components/data_frame/data_type/test_df_data_type.py

@@ -25,3 +25,18 @@ def test_data_frame_data_type(
     controller.OutputCode(page, "nw_data_view_selected").expect_value(
         re.compile(r"narwhals")
     )
+
+    controller.OutputCode(page, "pd_type").expect_value(re.compile(r"pandas"))
+    controller.OutputCode(page, "pd_data").expect_value(re.compile(r"pandas"))
+    controller.OutputCode(page, "pd_data_view").expect_value(re.compile(r"pandas"))
+    controller.OutputCode(page, "pd_data_view_selected").expect_value(
+        re.compile(r"pandas")
+    )
+
+    # modin tests
+    controller.OutputCode(page, "mpd_type").expect_value(re.compile(r"modin"))
+    controller.OutputCode(page, "mpd_data").expect_value(re.compile(r"modin"))
+    controller.OutputCode(page, "mpd_data_view").expect_value(re.compile(r"modin"))
+    controller.OutputCode(page, "mpd_data_view_selected").expect_value(
+        re.compile(r"modin")
+    )