UPDATE

2026-05-09 16:55:06 +03:30
parent 1679825ae2
commit cead7dafe2
51 changed files with 7514 additions and 1221 deletions
@@ -0,0 +1,135 @@
+from datetime import date
+
+from django.test import TestCase
+
+from location_data.data_driven_subdivision import sync_block_subdivision_with_result
+from location_data.models import (
+    AnalysisGridCell,
+    AnalysisGridObservation,
+    BlockSubdivision,
+    RemoteSensingRun,
+    RemoteSensingSubdivisionResult,
+    SoilLocation,
+)
+
+
+class DataDrivenSubdivisionSyncTests(TestCase):
+    def setUp(self):
+        self.boundary = {
+            "type": "Polygon",
+            "coordinates": [
+                [
+                    [51.3890, 35.6890],
+                    [51.3900, 35.6890],
+                    [51.3900, 35.6900],
+                    [51.3890, 35.6900],
+                    [51.3890, 35.6890],
+                ]
+            ],
+        }
+        self.location = SoilLocation.objects.create(
+            latitude="35.689200",
+            longitude="51.389000",
+            farm_boundary=self.boundary,
+        )
+        self.subdivision = BlockSubdivision.objects.create(
+            soil_location=self.location,
+            block_code="block-1",
+            source_boundary=self.boundary,
+            chunk_size_sqm=900,
+            status="defined",
+        )
+        self.run = RemoteSensingRun.objects.create(
+            soil_location=self.location,
+            block_subdivision=self.subdivision,
+            block_code="block-1",
+            chunk_size_sqm=900,
+            temporal_start=date(2025, 1, 1),
+            temporal_end=date(2025, 1, 31),
+            status=RemoteSensingRun.STATUS_SUCCESS,
+        )
+
+    def test_sync_block_subdivision_with_result_updates_saved_sub_blocks(self):
+        cell_1 = AnalysisGridCell.objects.create(
+            soil_location=self.location,
+            block_subdivision=self.subdivision,
+            block_code="block-1",
+            cell_code="cell-1",
+            chunk_size_sqm=900,
+            geometry=self.boundary,
+            centroid_lat="35.689200",
+            centroid_lon="51.389200",
+        )
+        cell_2 = AnalysisGridCell.objects.create(
+            soil_location=self.location,
+            block_subdivision=self.subdivision,
+            block_code="block-1",
+            cell_code="cell-2",
+            chunk_size_sqm=900,
+            geometry=self.boundary,
+            centroid_lat="35.689700",
+            centroid_lon="51.389700",
+        )
+        observation_1 = AnalysisGridObservation.objects.create(
+            cell=cell_1,
+            run=self.run,
+            temporal_start=date(2025, 1, 1),
+            temporal_end=date(2025, 1, 31),
+            ndvi=0.5,
+        )
+        observation_2 = AnalysisGridObservation.objects.create(
+            cell=cell_2,
+            run=self.run,
+            temporal_start=date(2025, 1, 1),
+            temporal_end=date(2025, 1, 31),
+            ndvi=0.7,
+        )
+        result = RemoteSensingSubdivisionResult.objects.create(
+            soil_location=self.location,
+            run=self.run,
+            block_subdivision=self.subdivision,
+            block_code="block-1",
+            chunk_size_sqm=900,
+            temporal_start=date(2025, 1, 1),
+            temporal_end=date(2025, 1, 31),
+            cluster_count=2,
+            selected_features=["ndvi"],
+            metadata={
+                "used_cell_count": 2,
+                "skipped_cell_count": 0,
+                "inertia_curve": [{"k": 1, "sse": 1.0}, {"k": 2, "sse": 0.1}],
+            },
+        )
+
+        sync_block_subdivision_with_result(
+            block_subdivision=self.subdivision,
+            result=result,
+            observations=[observation_1, observation_2],
+            cluster_summaries=[
+                {
+                    "cluster_label": 0,
+                    "centroid_lat": 35.6892,
+                    "centroid_lon": 51.3892,
+                    "cell_count": 1,
+                    "cell_codes": ["cell-1"],
+                },
+                {
+                    "cluster_label": 1,
+                    "centroid_lat": 35.6897,
+                    "centroid_lon": 51.3897,
+                    "cell_count": 1,
+                    "cell_codes": ["cell-2"],
+                },
+            ],
+        )
+
+        self.subdivision.refresh_from_db()
+        self.assertEqual(self.subdivision.status, "subdivided")
+        self.assertEqual(self.subdivision.grid_point_count, 2)
+        self.assertEqual(self.subdivision.centroid_count, 2)
+        self.assertEqual(self.subdivision.grid_points[0]["cell_code"], "cell-1")
+        self.assertEqual(self.subdivision.centroid_points[0]["sub_block_code"], "cluster-0")
+        self.assertEqual(
+            self.subdivision.metadata["data_driven_subdivision"]["cluster_count"],
+            2,
+        )