Add test for duckdb table with no property columns (#623)

Closes #622

Author: kylebarron

lonboard/_layer.py

@@ -1971,7 +1971,7 @@ class HeatmapLayer(BaseArrowLayer):
     def __init__(
         self, *, table: ArrowStreamExportable, **kwargs: Unpack[HeatmapLayerKwargs]
     ):
-        err_msg = """\
+        err_msg = """
         The `HeatmapLayer` is not currently working.
 
         As of Lonboard v0.10, Lonboard upgraded to version 9.0 of the underlying

lonboard/_viz.py

@@ -192,14 +192,13 @@ def viz(
         widget visualizing the provided data.
     """
     global COLOR_COUNTER
-    color_ordering = COLORS.copy()
 
     if isinstance(data, (list, tuple)):
         layers: List[Union[ScatterplotLayer, PathLayer, PolygonLayer]] = []
         for i, item in enumerate(data):
             ls = create_layers_from_data_input(
                 item,
-                _viz_color=color_ordering[(COLOR_COUNTER + i) % len(color_ordering)],
+                _viz_color=COLORS[(COLOR_COUNTER + i) % len(COLORS)],
                 scatterplot_kwargs=scatterplot_kwargs,
                 path_kwargs=path_kwargs,
                 polygon_kwargs=polygon_kwargs,
@@ -211,7 +210,7 @@ def viz(
     else:
         layers = create_layers_from_data_input(
             data,
-            _viz_color=color_ordering[COLOR_COUNTER % len(color_ordering)],
+            _viz_color=COLORS[COLOR_COUNTER % len(COLORS)],
             scatterplot_kwargs=scatterplot_kwargs,
             path_kwargs=path_kwargs,
             polygon_kwargs=polygon_kwargs,

poetry.lock

@@ -54,7 +54,7 @@ cli = ["click", "pyogrio", "shapely"]
 geopandas = ["geopandas", "pandas", "shapely"]
 
 [tool.poetry.group.dev.dependencies]
-duckdb = "^0.10.2"
+duckdb = ">=0.10.2"
 geoarrow-pyarrow = "^0.1.1"
 geoarrow-rust-core = "^0.2.0"
 geodatasets = "^2023.12.0"

pyproject.toml

@@ -205,3 +205,20 @@ def test_create_table_as_custom_con():
     # Succeeds when passing in con object
     m = viz(con.table("test"), con=con)
     assert isinstance(m.layers[0], ScatterplotLayer)
+
+
+def test_geometry_only_column():
+    # https://github.com/developmentseed/lonboard/issues/622
+    con = duckdb.connect()
+    sql = f"""
+        INSTALL spatial;
+        LOAD spatial;
+        CREATE TABLE data AS
+            SELECT CAST(geom as POINT_2D) as geom FROM ST_Read("{cities_gdal_path}");
+        """
+    con.execute(sql)
+
+    _layer = ScatterplotLayer.from_duckdb(con.table("data"), con)
+
+    m = viz(con.table("data"), con=con)
+    assert isinstance(m.layers[0], ScatterplotLayer)