added Feedback factory

hotosm · Apr 2, 2024 · 450db06 · 450db06
1 parent af30aca
commit 450db06
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/backend/tests/factories.py b/backend/tests/factories.py
@@ -1,7 +1,7 @@
 import factory
 
 from login.models import OsmUser
-from core.models import Dataset, AOI, Label, Model, Training
+from core.models import Dataset, AOI, Label, Model, Training, Feedback
 
 
 class OsmUserFactory(factory.django.DjangoModelFactory):
@@ -61,3 +61,16 @@ class Meta:
     created_by = factory.SubFactory(OsmUserFactory)
     epochs = 3
     batch_size = 24
+
+
+class FeedbackFactory(factory.django.DjangoModelFactory):
+
+    class Meta:
+        model = Feedback
+
+    geom = "POLYGON ((0 0, 0 1, 1 1, 1 0, 0 0))"
+    training = factory.SubFactory(TrainingFactory)
+    zoom_level = 19
+    feedback_type = "TP"
+    user = factory.SubFactory(OsmUserFactory)
+    source_imagery = "https://test_data/hotosm/fAIr/"