test: refactor test_data_color_utils.py through parametrization

great_tables/_tbl_data.py

@@ -29,10 +29,10 @@
     PlSelectExpr = _selector_proxy_
     PlExpr = pl.Expr
 
-    PdSeries = pd.Series
+    PdSeries = pd.Series[Any]
     PlSeries = pl.Series
-    PyArrowArray = pa.Array
-    PyArrowChunkedArray = pa.ChunkedArray
+    PyArrowArray = pa.Array[Any]
+    PyArrowChunkedArray = pa.ChunkedArray[Any]
 
     PdNA = pd.NA
     PlNull = pl.Null
@@ -759,7 +759,7 @@ def _(df: PyArrowTable, x: Any) -> bool:
     import pyarrow as pa
 
     arr = pa.array([x])
-    return arr.is_null().to_pylist()[0] or arr.is_nan().to_pylist()[0]
+    return arr.is_null(nan_is_null=True).to_pylist()[0]
 
 
 @singledispatch

tests/conftest.py

@@ -0,0 +1,69 @@
+from __future__ import annotations
+
+from importlib.util import find_spec
+
+import pytest
+
+from great_tables._tbl_data import DataFrameLike
+from tests.utils import DataFrameConstructor, DataLike
+
+
+def pandas_constructor(obj: DataLike) -> DataFrameLike:
+    import pandas as pd
+
+    return pd.DataFrame(obj)  # type: ignore[no-any-return]
+
+
+def pandas_nullable_constructor(obj: DataLike) -> DataFrameLike:
+    import pandas as pd
+
+    return pd.DataFrame(obj).convert_dtypes(dtype_backend="numpy_nullable")  # type: ignore[no-any-return]
+
+
+def pandas_pyarrow_constructor(obj: DataLike) -> DataFrameLike:
+    import pandas as pd
+
+    return pd.DataFrame(obj).convert_dtypes(dtype_backend="pyarrow")  # type: ignore[no-any-return]
+
+
+def polars_constructor(obj: DataLike) -> DataFrameLike:
+    import polars as pl
+
+    return pl.DataFrame(obj)
+
+
+def pyarrow_table_constructor(obj: DataLike) -> DataFrameLike:
+    import pyarrow as pa
+
+    return pa.table(obj)  # type: ignore[no-any-return]
+
+
+frame_constructors: list[DataFrameConstructor] = []
+
+is_pyarrow_installed = find_spec("pyarrow") is not None
+
+if find_spec("pandas"):
+    import pandas as pd
+
+    frame_constructors.append(pandas_constructor)
+
+    pandas_ge_v2 = pd.__version__ >= "2.0.0"
+
+    if pandas_ge_v2:
+        frame_constructors.append(pandas_nullable_constructor)
+
+    if pandas_ge_v2 and is_pyarrow_installed:
+        # pandas 2.0+ supports pyarrow dtype backend
+        # https://pandas.pydata.org/docs/whatsnew/v2.0.0.html#new-dtype-backends
+        frame_constructors.append(pandas_pyarrow_constructor)
+
+if find_spec("polars"):
+    frame_constructors.append(polars_constructor)
+
+if is_pyarrow_installed:
+    frame_constructors.append(pyarrow_table_constructor)
+
+
+@pytest.fixture(params=frame_constructors)
+def frame_constructor(request: pytest.FixtureRequest) -> DataFrameConstructor:
+    return request.param  # type: ignore[no-any-return]