tests(dataframe): Add additional backends to test dataframe compatibi…

…lity (#1723)

pyproject.toml

@@ -47,7 +47,7 @@ dependencies = [
     "prompt-toolkit;platform_system!='Emscripten'",
     "python-multipart>=0.0.7;platform_system!='Emscripten'",
     "setuptools;python_version>='3.12'",
-    "narwhals>=1.9.1",
+    "narwhals>=1.9.3",
     "orjson>=3.10.7",
 ]
 
@@ -87,6 +87,11 @@ test = [
     "faicons",
     "ridgeplot",
     "great_tables",
+    "modin[all]",
+    "polars",
+    "dask[dataframe]",
+    "pyarrow",
+    "pyarrow-stubs",
 ]
 dev = [
     "black>=24.0",

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

@@ -1,23 +1,38 @@
 from __future__ import annotations
 
+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
 
 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(pd_df)  # pyright: ignore[reportUnknownMemberType]
+mpd_df = mpd.DataFrame(pd_df)
+pl_df = pl.DataFrame(pd_df)
+
 
 with ui.layout_columns():
 
     with ui.card():
-
         ui.h2("Original Pandas Data")
 
         @render.data_frame
         def pd_df_original():
-            return pd_df
+            return render.DataGrid(
+                data=pd_df,  # pyright: ignore[reportUnknownArgumentType]
+                selection_mode="row",
+            )
+
+        "Selected row:"
+
+        @render.data_frame
+        def selected_pandas_row():
+            return pd_df_original.data_view(selected=True)
 
         "Data type:"
 
@@ -48,7 +63,16 @@ def pd_data_view_selected():
 
         @render.data_frame
         def nw_df_original():
-            return nw_df
+            return render.DataGrid(
+                data=nw_df,
+                selection_mode="row",
+            )
+
+        "Selected row:"
+
+        @render.data_frame
+        def selected_nw_row():
+            return nw_df_original.data_view(selected=True)
 
         "Data type:"
 
@@ -73,3 +97,123 @@ 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 render.DataGrid(
+                data=pa_df,
+                selection_mode="row",
+            )
+
+        "Selected row:"
+
+        @render.data_frame
+        def selected_pa_row():
+            return pa_df_original.data_view(selected=True)
+
+        "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 render.DataGrid(
+                data=mpd_df,
+                selection_mode="row",
+            )
+
+        "Selected row:"
+
+        @render.data_frame
+        def selected_mpd_row():
+            return mpd_df_original.data_view(selected=True)
+
+        "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 render.DataGrid(
+                data=pl_df,
+                selection_mode="row",
+            )
+
+        "Selected row:"
+
+        @render.data_frame
+        def selected_pl_row():
+            return pl_df_original.data_view(selected=True)
+
+        "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)))

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

@@ -5,23 +5,64 @@
 from shiny.playwright import controller
 from shiny.run import ShinyAppProc
 
+backends = [
+    {
+        "name": "pandas",
+        "prefix": "pd",
+        "df_original": "pd_df_original",
+        "selected_row": "selected_pandas_row",
+    },
+    {
+        "name": "narwhals",
+        "prefix": "nw",
+        "df_original": "nw_df_original",
+        "selected_row": "selected_nw_row",
+    },
+    {
+        "name": "pyarrow",
+        "prefix": "pa",
+        "df_original": "pa_df_original",
+        "selected_row": "selected_pa_row",
+    },
+    {
+        "name": "polars",
+        "prefix": "pl",
+        "df_original": "pl_df_original",
+        "selected_row": "selected_pl_row",
+    },
+    {
+        "name": "modin",
+        "prefix": "mpd",
+        "df_original": "mpd_df_original",
+        "selected_row": "selected_mpd_row",
+    },
+]
+
 
 def test_data_frame_data_type(
     page: Page,
     local_app: ShinyAppProc,
 ) -> None:
     page.goto(local_app.url)
+    # Iterate over the backends
+    for backend in backends:
+        # Perform output code tests
+        controller.OutputCode(page, f"{backend['prefix']}_type").expect_value(
+            re.compile(backend["name"])
+        )
+        controller.OutputCode(page, f"{backend['prefix']}_data").expect_value(
+            re.compile(backend["name"])
+        )
+        controller.OutputCode(page, f"{backend['prefix']}_data_view").expect_value(
+            re.compile(backend["name"])
+        )
+        controller.OutputCode(
+            page, f"{backend['prefix']}_data_view_selected"
+        ).expect_value(re.compile(backend["name"]))
 
-    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")
-    )
-
-    controller.OutputCode(page, "nw_type").expect_value(re.compile(r"narwhals"))
-    controller.OutputCode(page, "nw_data").expect_value(re.compile(r"narwhals"))
-    controller.OutputCode(page, "nw_data_view").expect_value(re.compile(r"narwhals"))
-    controller.OutputCode(page, "nw_data_view_selected").expect_value(
-        re.compile(r"narwhals")
-    )
+        # Perform output dataframe tests
+        controller.OutputDataFrame(page, backend["selected_row"]).expect_column_labels(
+            ["studyName", "Sample Number"]
+        )
+        controller.OutputDataFrame(page, backend["df_original"]).select_rows([1])
+        controller.OutputDataFrame(page, backend["selected_row"]).expect_nrow(1)

tests/playwright/shiny/components/table/app.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import modin.pandas as md  # pyright: ignore[reportMissingTypeStubs]
 import narwhals.stable.v1 as nw
 import palmerpenguins  # pyright: ignore[reportMissingTypeStubs]
 
@@ -9,6 +10,8 @@
 
 nw_df = nw.from_native(pd_df, eager_only=True)
 
+md_df = md.DataFrame(pd_df)
+
 with ui.card():
 
     ui.h2("Polars Pandas Data")
@@ -22,3 +25,18 @@ def nw_table():
     @render.code
     def nw_df_type():
         return str(type(nw_df))
+
+
+with ui.card():
+
+    ui.h2("Modin dataframe Data")
+
+    @render.table
+    def md_table():
+        return md_df
+
+    "Data type:"
+
+    @render.code
+    def md_df_type():
+        return str(type(md_df))

tests/playwright/shiny/components/table/test_table.py

@@ -9,8 +9,8 @@
 def test_table_data_support(page: Page, local_app: ShinyAppProc) -> None:
     page.goto(local_app.url)
 
-    table = controller.OutputTable(page, "nw_table")
-
-    table.expect_nrow(2)
-
+    controller.OutputTable(page, "nw_table").expect_nrow(2)
     controller.OutputCode(page, "nw_df_type").expect_value(re.compile("narwhals"))
+
+    controller.OutputTable(page, "md_table").expect_nrow(2)
+    controller.OutputCode(page, "md_df_type").expect_value(re.compile("modin"))