Prepare some tests for Ibis

Author: mwouts

.github/workflows/continuous-integration.yml

@@ -50,6 +50,8 @@ jobs:
           - python-version: "3.11"
             pandas-version: pre
             polars: true
+          - python-version: "3.11"
+            ibis: true
           - python-version: "3.11"
             uninstall_jinja2: true
     runs-on: ubuntu-20.04
@@ -85,6 +87,10 @@ jobs:
         if: matrix.polars
         run: pip install -e .[polars]
 
+      - name: Install Ibis
+        if: matrix.ibis
+        run: pip install -e .[ibis]
+
       - name: Uninstall jinja2
         if: matrix.uninstall_jinja2
         run: pip uninstall jinja2 -y

itables/sample_dfs.py

@@ -105,7 +105,7 @@ def get_df_complex_index():
     return df
 
 
-def get_dict_of_test_dfs(N=100, M=100, polars=False):
+def get_dict_of_test_dfs(N=100, M=100, polars=False, ibis=False):
     NM_values = np.reshape(np.linspace(start=0.0, stop=1.0, num=N * M), (N, M))
 
     test_dfs = {
@@ -279,6 +279,19 @@ def get_dict_of_test_dfs(N=100, M=100, polars=False):
                 pass
         return polars_dfs
 
+    if ibis:
+        import ibis as ib
+
+        con = ib.pandas.connect(test_dfs)
+        ibis_dfs = {}
+        for key in test_dfs:
+            try:
+                ibis_dfs[key] = con.table(key)
+            except (TypeError, AttributeError):
+                pass
+
+        return ibis_dfs
+
     return test_dfs
 
 

setup.py

@@ -52,7 +52,10 @@
     },
     tests_require=["pytest", "pytz"],
     install_requires=["IPython", "pandas", "numpy"],
-    extras_require={"polars": ["polars", "pyarrow"]},
+    extras_require={
+        "polars": ["polars", "pyarrow"],
+        "ibis": ["ibis-framework[pandas]"],
+    },
     license="MIT",
     classifiers=[
         "Development Status :: 5 - Production/Stable",

tests/test_ibis.py

@@ -0,0 +1,16 @@
+import pytest
+
+from itables import to_html_datatable
+from itables.sample_dfs import get_dict_of_test_dfs
+
+try:
+    import ibis  # noqa
+except ImportError as e:
+    pytest.skip(str(e), allow_module_level=True)
+
+
+@pytest.mark.parametrize(
+    "name,df", [(name, df) for name, df in get_dict_of_test_dfs(ibis=True).items()]
+)
+def test_show_ibis_df(name, df, use_to_html):
+    to_html_datatable(df, use_to_html)