Add tests for points_to_geojson

mkdocs.yml

@@ -20,6 +20,7 @@ nav:
         - Read Geo-tif Data: read_geotif.md
         - Transform coordinate points: transform_points.md
         - Transform coordinate polygons: transform_polygons.md
+        - Save clicked points as geojson: points_to_geojson.md
 markdown_extensions:
   - toc:
       permalink: True

pyproject.toml

@@ -14,6 +14,7 @@ dependencies = [
     "rasterio",
     "fiona",
     "shapely",
+    "plantcv.annotate",
 ]
 requires-python = ">=3.6"
 authors = [

tests/test_geospatial_points_to_geojson.py

@@ -0,0 +1,34 @@
+"""Tests for geospatial.points_to_geojson"""
+
+import pytest
+import plantcv.annotate as an
+from plantcv.geospatial import read_geotif
+from plantcv.geospatial import points_to_geojson
+
+def test_geospatial_points_to_geojson_napari(test_data, tmpdir):
+    """Test for plantcv-geospatial."""
+    cache_dir = tmpdir.mkdir("cache")
+    img = read_geotif(filename=test_data.rgb_tif, bands="R,G,B")
+    viewer = an.napari_open(img=img.pseudo_rgb, show=False)
+    viewer.add_points()
+    filename = os.path.join(cache_dir, 'test_out.geojson')
+    points_to_geojson(img, viewer, output=filename)
+    assert os.path.exists(filename)
+
+def test_geospatial_points_to_geojson_an(test_data, tmpdir):
+    """Test for plantcv-geospatial."""
+    cache_dir = tmpdir.mkdir("cache")
+    img = read_geotif(filename=test_data.rgb_tif, bands="R,G,B")
+    viewer = an.Points(img=img.pseudo_rgb)
+    filename = os.path.join(cache_dir, 'test_out.geojson')
+    points_to_geojson(img, viewer, output=filename)
+    assert os.path.exists(filename)
+
+def test_geospatial_points_to_geojson_badviewer(test_data, tmpdir):
+    """Test for plantcv-geospatial."""
+    cache_dir = tmpdir.mkdir("cache")
+    img = read_geotif(filename=test_data.rgb_tif, bands="R,G,B")
+    viewer = []
+    filename = os.path.join(cache_dir, 'test_out.geojson')
+    with pytest.raises(RuntimeError):
+        points_to_geojson(img, viewer, output=filename)