summaryrefslogtreecommitdiff
path: root/tests/gis_tests
diff options
context:
space:
mode:
authorMariusz Felisiak <felisiak.mariusz@gmail.com>2024-06-08 10:19:55 +0200
committerSarah Boyce <42296566+sarahboyce@users.noreply.github.com>2024-06-18 12:03:57 +0200
commita0c44d4e23f8f509757f97f28fbbb1ced3382361 (patch)
treea8a4fcf96f708c1e1706e548b72ebf88f2ee5696 /tests/gis_tests
parent1b21feeb7b490b3c75a06736362b05251ec172a9 (diff)
Simplified OperationTestCase.alter_gis_model() test hook a bit.
This avoids passing "blank=False" and "srid=4326" to field classes, which are the default values, and removes special treatment for the "blank" parameter.
Diffstat (limited to 'tests/gis_tests')
-rw-r--r--tests/gis_tests/gis_migrations/test_operations.py31
1 files changed, 16 insertions, 15 deletions
diff --git a/tests/gis_tests/gis_migrations/test_operations.py b/tests/gis_tests/gis_migrations/test_operations.py
index e81d44caf2..033f2ac5b9 100644
--- a/tests/gis_tests/gis_migrations/test_operations.py
+++ b/tests/gis_tests/gis_migrations/test_operations.py
@@ -97,13 +97,12 @@ class OperationTestCase(TransactionTestCase):
migration_class,
model_name,
field_name,
- blank=False,
field_class=None,
field_class_kwargs=None,
):
args = [model_name, field_name]
if field_class:
- field_class_kwargs = field_class_kwargs or {"srid": 4326, "blank": blank}
+ field_class_kwargs = field_class_kwargs or {}
args.append(field_class(**field_class_kwargs))
operation = migration_class(*args)
old_state = self.current_state.clone()
@@ -122,7 +121,7 @@ class OperationTests(OperationTestCase):
Test the AddField operation with a geometry-enabled column.
"""
self.alter_gis_model(
- migrations.AddField, "Neighborhood", "path", False, fields.LineStringField
+ migrations.AddField, "Neighborhood", "path", fields.LineStringField
)
self.assertColumnExists("gis_neighborhood", "path")
@@ -147,7 +146,7 @@ class OperationTests(OperationTestCase):
Test the AddField operation with a raster-enabled column.
"""
self.alter_gis_model(
- migrations.AddField, "Neighborhood", "heatmap", False, fields.RasterField
+ migrations.AddField, "Neighborhood", "heatmap", fields.RasterField
)
self.assertColumnExists("gis_neighborhood", "heatmap")
@@ -160,7 +159,11 @@ class OperationTests(OperationTestCase):
Should be able to add a GeometryField with blank=True.
"""
self.alter_gis_model(
- migrations.AddField, "Neighborhood", "path", True, fields.LineStringField
+ migrations.AddField,
+ "Neighborhood",
+ "path",
+ fields.LineStringField,
+ field_class_kwargs={"blank": True},
)
self.assertColumnExists("gis_neighborhood", "path")
@@ -178,7 +181,11 @@ class OperationTests(OperationTestCase):
Should be able to add a RasterField with blank=True.
"""
self.alter_gis_model(
- migrations.AddField, "Neighborhood", "heatmap", True, fields.RasterField
+ migrations.AddField,
+ "Neighborhood",
+ "heatmap",
+ fields.RasterField,
+ field_class_kwargs={"blank": True},
)
self.assertColumnExists("gis_neighborhood", "heatmap")
@@ -247,9 +254,8 @@ class OperationTests(OperationTestCase):
migrations.AlterField,
"Neighborhood",
"geom",
- False,
fields.MultiPolygonField,
- field_class_kwargs={"srid": 4326, "dim": 3},
+ field_class_kwargs={"dim": 3},
)
self.assertTrue(Neighborhood.objects.first().geom.hasz)
# Rewind to 2 dimensions.
@@ -257,9 +263,8 @@ class OperationTests(OperationTestCase):
migrations.AlterField,
"Neighborhood",
"geom",
- False,
fields.MultiPolygonField,
- field_class_kwargs={"srid": 4326, "dim": 2},
+ field_class_kwargs={"dim": 2},
)
self.assertFalse(Neighborhood.objects.first().geom.hasz)
@@ -296,9 +301,5 @@ class NoRasterSupportTests(OperationTestCase):
with self.assertRaisesMessage(ImproperlyConfigured, msg):
self.set_up_test_model()
self.alter_gis_model(
- migrations.AddField,
- "Neighborhood",
- "heatmap",
- False,
- fields.RasterField,
+ migrations.AddField, "Neighborhood", "heatmap", fields.RasterField
)