Fix vector feature imports

OpenGeoscience · Jun 24, 2024 · 11d376c · 11d376c
1 parent d854226
commit 11d376c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/uvdat/core/models/__init__.py b/uvdat/core/models/__init__.py
@@ -2,7 +2,7 @@
 from .context import Context
 from .dataset import Dataset
 from .file_item import FileItem
-from .map_layers import RasterMapLayer, VectorMapLayer
+from .map_layers import RasterMapLayer, VectorFeature, VectorMapLayer
 from .networks import NetworkEdge, NetworkNode
 from .regions import DerivedRegion, SourceRegion
 from .simulations import SimulationResult
@@ -14,6 +14,7 @@
     FileItem,
     RasterMapLayer,
     VectorMapLayer,
+    VectorFeature,
     SourceRegion,
     DerivedRegion,
     NetworkEdge,

diff --git a/uvdat/core/tests/test_populate.py b/uvdat/core/tests/test_populate.py
@@ -12,6 +12,7 @@
     RasterMapLayer,
     SimulationResult,
     SourceRegion,
+    VectorFeature,
     VectorMapLayer,
 )
 
@@ -42,4 +43,4 @@ def test_populate():
     assert SimulationResult.objects.all().count() == 0
     assert SourceRegion.objects.all().count() == 24
     assert VectorMapLayer.objects.all().count() == 5
-    # TODO: Add vectorfeature assertion
+    assert VectorFeature.objects.count() > 0