115 lines
4.0 KiB
Python
115 lines
4.0 KiB
Python
from django.test import TestCase, override_settings
|
|
|
|
from location_data.grid_analysis import create_or_get_analysis_grid_cells
|
|
from location_data.models import AnalysisGridCell, BlockSubdivision, SoilLocation
|
|
|
|
|
|
@override_settings(SUBDIVISION_CHUNK_SQM=900)
|
|
class AnalysisGridServiceTests(TestCase):
|
|
def setUp(self):
|
|
self.boundary = {
|
|
"type": "Polygon",
|
|
"coordinates": [
|
|
[
|
|
[51.389000, 35.689000],
|
|
[51.389760, 35.689000],
|
|
[51.389760, 35.689620],
|
|
[51.389000, 35.689620],
|
|
[51.389000, 35.689000],
|
|
]
|
|
],
|
|
}
|
|
self.location = SoilLocation.objects.create(
|
|
latitude="35.689310",
|
|
longitude="51.389380",
|
|
farm_boundary=self.boundary,
|
|
)
|
|
self.location.set_input_block_count(1)
|
|
self.location.save(update_fields=["input_block_count", "block_layout", "updated_at"])
|
|
self.subdivision = BlockSubdivision.objects.create(
|
|
soil_location=self.location,
|
|
block_code="block-1",
|
|
source_boundary=self.boundary,
|
|
chunk_size_sqm=900,
|
|
status="created",
|
|
)
|
|
|
|
def test_create_analysis_grid_cells_persists_30x30_cells(self):
|
|
result = create_or_get_analysis_grid_cells(
|
|
self.location,
|
|
block_code="block-1",
|
|
block_subdivision=self.subdivision,
|
|
)
|
|
|
|
self.assertTrue(result["created"])
|
|
self.assertEqual(result["chunk_size_sqm"], 900)
|
|
self.assertGreater(result["created_count"], 0)
|
|
self.assertEqual(result["created_count"], result["total_count"])
|
|
|
|
cells = list(
|
|
AnalysisGridCell.objects.filter(
|
|
soil_location=self.location,
|
|
block_code="block-1",
|
|
chunk_size_sqm=900,
|
|
).order_by("cell_code")
|
|
)
|
|
self.assertEqual(len(cells), result["total_count"])
|
|
self.assertTrue(all(cell.block_subdivision_id == self.subdivision.id for cell in cells))
|
|
self.assertTrue(all(cell.geometry.get("type") == "Polygon" for cell in cells))
|
|
self.assertTrue(all(len(cell.geometry.get("coordinates", [[]])[0]) == 5 for cell in cells))
|
|
|
|
self.subdivision.refresh_from_db()
|
|
self.location.refresh_from_db()
|
|
self.assertEqual(
|
|
self.subdivision.metadata["analysis_grid"]["chunk_size_sqm"],
|
|
900,
|
|
)
|
|
self.assertEqual(
|
|
self.subdivision.metadata["analysis_grid"]["cell_count"],
|
|
result["total_count"],
|
|
)
|
|
self.assertEqual(
|
|
self.location.block_layout["blocks"][0]["analysis_grid_summary"]["chunk_size_sqm"],
|
|
900,
|
|
)
|
|
|
|
def test_create_analysis_grid_cells_is_idempotent(self):
|
|
first = create_or_get_analysis_grid_cells(
|
|
self.location,
|
|
block_code="block-1",
|
|
block_subdivision=self.subdivision,
|
|
)
|
|
second = create_or_get_analysis_grid_cells(
|
|
self.location,
|
|
block_code="block-1",
|
|
block_subdivision=self.subdivision,
|
|
)
|
|
|
|
self.assertTrue(first["created"])
|
|
self.assertFalse(second["created"])
|
|
self.assertEqual(second["created_count"], 0)
|
|
self.assertEqual(second["existing_count"], first["total_count"])
|
|
self.assertEqual(
|
|
AnalysisGridCell.objects.filter(
|
|
soil_location=self.location,
|
|
block_code="block-1",
|
|
chunk_size_sqm=900,
|
|
).count(),
|
|
first["total_count"],
|
|
)
|
|
|
|
def test_create_analysis_grid_cells_uses_location_boundary_without_subdivision(self):
|
|
result = create_or_get_analysis_grid_cells(
|
|
self.location,
|
|
block_code="",
|
|
)
|
|
|
|
self.assertGreater(result["total_count"], 0)
|
|
self.assertTrue(
|
|
AnalysisGridCell.objects.filter(
|
|
soil_location=self.location,
|
|
block_code="",
|
|
chunk_size_sqm=900,
|
|
).exists()
|
|
)
|