convert to polars if is list

Author: tylerriccio33

great_tables/_formats_vals.py

@@ -38,8 +38,17 @@ def _make_one_col_table(vals: X) -> GT:
         # anticipating a tuple may be too defensive
         vals = list(vals)
 
+    if isinstance(vals, list):
+        try:
+            import polars as pl
+        except ImportError:
+            pass
+        else:
+            vals = pl.Series(vals)
+
     # TODO: remove pandas. if vals is not a SeriesLike, then we currently
     # convert them to a pandas Series for backwards compatibility.
+
     df = to_frame(vals, name="x")
 
     # Convert the list to a Pandas DataFrame and then to a GTData object

tests/test__formats_vals.py

@@ -3,10 +3,26 @@
 import pytest
 from great_tables._formats_vals import _make_one_col_table
 from great_tables._tbl_data import to_list
+import sys
 
 
 @pytest.mark.parametrize("src", [1, [1], (1,), pd.Series([1]), pl.Series([1])])
 def test_roundtrip(src):
     gt = _make_one_col_table(src)
 
     assert to_list(gt._tbl_data["x"]) == [1]
+
+
+def test_one_column_table_no_lib() -> None:
+    sys.modules["pandas"] = None
+
+    with pytest.raises(ModuleNotFoundError):
+        import pandas
+
+    vals = [1, 2, 3]
+
+    _make_one_col_table(vals=vals)
+    raise
+
+
+test_one_column_table_no_lib()