Tests/training-model

backend/tests/factories.py

@@ -0,0 +1,63 @@
+import factory
+
+from login.models import OsmUser
+from core.models import Dataset, AOI, Label, Model, Training
+
+
+class OsmUserFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = OsmUser
+
+    username = "Test User"
+    osm_id = 123456
+
+
+class DatasetFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = Dataset
+
+    name = "Test Dataset"
+    created_by = factory.SubFactory(OsmUserFactory)
+    status = -1
+
+
+class AoiFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = AOI
+
+    geom = "POLYGON ((0 0, 0 1, 1 1, 1 0, 0 0))"
+    dataset = factory.SubFactory(DatasetFactory)
+    label_status = -1
+
+
+class LabelFactory(factory.django.DjangoModelFactory):
+
+    class Meta:
+        model = Label
+
+    aoi = factory.SubFactory(AoiFactory)
+    geom = "POLYGON ((0 0, 0 1, 1 1, 1 0, 0 0))"
+
+
+class ModelFactory(factory.django.DjangoModelFactory):
+
+    class Meta:
+        model = Model
+
+    dataset = factory.SubFactory(DatasetFactory)
+    name = "Test Model"
+    created_by = factory.SubFactory(OsmUserFactory)
+    status = -1
+
+
+class TrainingFactory(factory.django.DjangoModelFactory):
+
+    class Meta:
+        model = Training
+
+    model = factory.SubFactory(ModelFactory)
+    status = "SUBMITTED"
+    zoom_level = [20, 21]
+    created_by = factory.SubFactory(OsmUserFactory)
+    epochs = 3
+    batch_size = 24

backend/tests/test_core_model.py

@@ -0,0 +1,46 @@
+from django.test import TestCase
+
+from .factories import (
+    DatasetFactory,
+    OsmUserFactory,
+    AoiFactory,
+    LabelFactory,
+    ModelFactory,
+    TrainingFactory,
+)
+
+
+class TestCoreModels(TestCase):
+
+    def setUp(self):
+        self.user = OsmUserFactory(username="Test User 2", osm_id=123)
+        self.dataset = DatasetFactory(created_by=self.user)
+        self.aoi = AoiFactory(dataset=self.dataset)
+        self.label = LabelFactory(aoi=self.aoi)
+        self.model = ModelFactory(dataset=self.dataset, created_by=self.user)
+        self.training = TrainingFactory(model=self.model, created_by=self.user)
+
+    def test_dataset_creation(self):
+        self.assertEqual(self.dataset.name, "Test Dataset")
+        self.assertEqual(self.dataset.created_by, self.user)
+
+    def test_aoi_creation(self):
+        self.assertEqual(self.aoi.dataset, self.dataset)
+        self.assertEqual(self.aoi.label_status, -1)
+
+    def test_label_creation(self):
+        self.assertEqual(self.label.aoi, self.aoi)
+
+    def test_model_creation(self):
+        self.assertEqual(self.model.name, "Test Model")
+        self.assertEqual(self.model.dataset, self.dataset)
+        self.assertEqual(self.model.created_by, self.user)
+        self.assertEqual(self.model.status, -1)
+
+    def test_training_creation(self):
+        self.assertEqual(self.training.model, self.model)
+        self.assertEqual(self.training.status, "SUBMITTED")
+        self.assertEqual(self.training.zoom_level, [20, 21])
+        self.assertEqual(self.training.created_by, self.user)
+        self.assertEqual(self.training.epochs, 3)
+        self.assertEqual(self.training.batch_size, 24)

backend/tests/test_login_model.py

@@ -0,0 +1,12 @@
+from django.test import TestCase
+
+from .factories import OsmUserFactory
+
+
+class TestLoginModels(TestCase):
+
+    def test_OsmUser_creation(self):
+        osm_user = OsmUserFactory()
+
+        self.assertEqual(str(osm_user), "Test User")
+        self.assertEqual(osm_user.osm_id, 123456)