Fix viz() with __geo_interface__ input (#426)

Closes #424

Author: kylebarron

lonboard/_viz.py

@@ -189,12 +189,12 @@ def create_layer_from_data_input(
 
     # Anything with __arrow_c_stream__
     if hasattr(data, "__arrow_c_stream__"):
-        data = cast(ArrowStreamExportable, data)
+        data = cast("ArrowStreamExportable", data)
         return _viz_geoarrow_table(pa.table(data), **kwargs)
 
     # Anything with __geo_interface__
     if hasattr(data, "__geo_interface__"):
-        data = cast(GeoInterfaceProtocol, data)
+        data = cast("GeoInterfaceProtocol", data)
         return _viz_geo_interface(data.__geo_interface__, **kwargs)
 
     # GeoJSON dict

tests/test_viz.py

@@ -13,11 +13,27 @@ def test_viz_wkb_geoarrow():
 
 
 def test_viz_reproject():
-    gdf = gpd.read_file(gpd.datasets.get_path("nybb"))
+    gdf = gpd.read_file(geodatasets.get_path("nybb"))
     map_ = viz(gdf)
 
     # Assert table was reprojected
     first_coord = map_.layers[0].table["geometry"][0][0][0][0]
     expected_coord = -74.05050951794041, 40.56643026026788
     assert (first_coord[0].as_py() - expected_coord[0]) < 0.0001
     assert (first_coord[1].as_py() - expected_coord[1]) < 0.0001
+
+
+def test_viz_geo_interface():
+    class GeoInterfaceHolder:
+        def __init__(self, geom) -> None:
+            self.geom = geom
+
+        @property
+        def __geo_interface__(self):
+            return self.geom.__geo_interface__
+
+    gdf = gpd.read_file(geodatasets.get_path("nybb")).to_crs("EPSG:4326")
+    geo_interface_obj = GeoInterfaceHolder(gdf.geometry[0])
+    map_ = viz(geo_interface_obj)
+
+    assert isinstance(map_.layers[0], SolidPolygonLayer)