UPDATE

location_data/test_remote_sensing_api.py

@@ -11,6 +11,7 @@ from location_data.data_driven_subdivision import DEFAULT_CLUSTER_FEATURES
 from location_data.models import (
     AnalysisGridCell,
     AnalysisGridObservation,
+    RemoteSensingClusterBlock,
     BlockSubdivision,
     RemoteSensingClusterAssignment,
     RemoteSensingRun,
@@ -457,6 +458,21 @@ class RemoteSensingApiTests(TestCase):
             raw_feature_values={"ndvi": 0.61},
             scaled_feature_values={"ndvi": 0.0},
         )
+        cluster_block = RemoteSensingClusterBlock.objects.create(
+            result=result,
+            soil_location=self.location,
+            block_subdivision=self.subdivision,
+            block_code="",
+            sub_block_code="cluster-0",
+            cluster_label=0,
+            chunk_size_sqm=900,
+            centroid_lat="35.689500",
+            centroid_lon="51.389500",
+            cell_count=1,
+            cell_codes=["cell-1"],
+            geometry=self.boundary,
+            metadata={"source": "analysis_grid_cells"},
+        )
 
         task_id = "e723ba3e-c53c-401b-b3a0-5f7013c7b401"
         run.metadata = {**run.metadata, "task_id": task_id}
@@ -468,5 +484,6 @@ class RemoteSensingApiTests(TestCase):
         payload = response.json()["data"]
         self.assertEqual(payload["status"], "completed")
         self.assertEqual(payload["subdivision_result"]["cluster_count"], 1)
+        self.assertEqual(payload["subdivision_result"]["cluster_blocks"][0]["uuid"], str(cluster_block.uuid))
         self.assertEqual(len(payload["subdivision_result"]["assignments"]), 1)
         self.assertEqual(payload["pagination"]["assignments"]["total_items"], 1)