def testFilter(self):
"""
Test filtering inside the dialog
"""
layer = make_scenario_layer()
mb_electorate_layer = make_meshblock_electorate_layer()
registry = ScenarioRegistry(
source_layer=layer,
id_field='id',
name_field='name',
meshblock_electorate_layer=mb_electorate_layer)
dlg = ScenarioSelectionDialog(scenario_registry=registry)
self.assertEqual(
[dlg.list.item(r).text() for r in range(dlg.list.count())],
['Scenario 1', 'scenario 3', 'scenario B'])
dlg.search.setText('eee') # connection not fired on first change?
dlg.search.setText('3')
self.assertEqual([
dlg.list.item(r).text() for r in range(dlg.list.count())
if not dlg.list.item(r).isHidden()
], ['scenario 3'])
dlg.search.setText('B')
self.assertEqual([
dlg.list.item(r).text() for r in range(dlg.list.count())
if not dlg.list.item(r).isHidden()
], ['scenario B'])
# case insensitive!
dlg.search.setText('b')
self.assertEqual([
dlg.list.item(r).text() for r in range(dlg.list.count())
if not dlg.list.item(r).isHidden()
], ['scenario B'])
def testConstruct(self):
"""
Test constructing dock
"""
scenario_layer = make_scenario_layer()
scenario_registry = ScenarioRegistry(source_layer=scenario_layer,
id_field='id',
name_field='name',
meshblock_electorate_layer=None)
context = LinzRedistrictingContext(scenario_registry=scenario_registry)
widget = LinzRedistrictingDockWidget(context=context, iface=IFACE)
self.assertIsNotNone(widget)
def testConstruct(self):
"""
Test creating dialog
"""
layer = make_scenario_layer()
mb_electorate_layer = make_meshblock_electorate_layer()
registry = ScenarioRegistry(
source_layer=layer,
id_field='id',
name_field='name',
meshblock_electorate_layer=mb_electorate_layer)
self.assertIsNotNone(
ScenarioSelectionDialog(scenario_registry=registry))
def testAccept(self):
"""
Test that accepting dialog
"""
layer = make_scenario_layer()
mb_electorate_layer = make_meshblock_electorate_layer()
registry = ScenarioRegistry(
source_layer=layer,
id_field='id',
name_field='name',
meshblock_electorate_layer=mb_electorate_layer)
dlg = ScenarioSelectionDialog(scenario_registry=registry)
dlg.set_selected_scenario('d4')
dlg.accept()
def testPopulation(self):
"""
Test that dialog is correctly populated from registry
"""
layer = make_scenario_layer()
mb_electorate_layer = make_meshblock_electorate_layer()
registry = ScenarioRegistry(
source_layer=layer,
id_field='id',
name_field='name',
meshblock_electorate_layer=mb_electorate_layer)
dlg = ScenarioSelectionDialog(scenario_registry=registry)
self.assertEqual(
[dlg.list.item(r).text() for r in range(dlg.list.count())],
['Scenario 1', 'scenario 3', 'scenario B'])
# initial selection must be final scenario
self.assertEqual(dlg.selected_scenario(), 2)
def testNameForCurrentTask(self):
"""
Test retrieving friendly name for currenttask
"""
scenario_layer = make_scenario_layer()
scenario_registry = ScenarioRegistry(source_layer=scenario_layer,
id_field='id',
name_field='name',
meshblock_electorate_layer=None)
context = LinzRedistrictingContext(scenario_registry=scenario_registry)
context.task = LinzRedistrictingContext.TASK_GN
self.assertEqual(context.get_name_for_current_task(),
'General (North Island)')
context.task = LinzRedistrictingContext.TASK_GS
self.assertEqual(context.get_name_for_current_task(),
'General (South Island)')
context.task = LinzRedistrictingContext.TASK_M
self.assertEqual(context.get_name_for_current_task(), 'Māori')
def testBasic(self):
"""
Test getters/settings
"""
scenario_layer = make_scenario_layer()
scenario_registry = ScenarioRegistry(source_layer=scenario_layer,
id_field='id',
name_field='name',
meshblock_electorate_layer=None)
context = LinzRedistrictingContext(scenario_registry=scenario_registry)
self.assertIn(context.task, (LinzRedistrictingContext.TASK_GN,
LinzRedistrictingContext.TASK_GS,
LinzRedistrictingContext.TASK_M))
context.task = LinzRedistrictingContext.TASK_GS
self.assertEqual(context.task, LinzRedistrictingContext.TASK_GS)
self.assertIsNotNone(context.scenario)
context.scenario = 10
self.assertEqual(context.scenario, 10)
context.set_scenario(3)
self.assertEqual(context.scenario, 3)
def testTitle(self):
"""
Test title updates
"""
scenario_layer = make_scenario_layer()
scenario_registry = ScenarioRegistry(source_layer=scenario_layer,
id_field='id',
name_field='name',
meshblock_electorate_layer=None)
context = LinzRedistrictingContext(scenario_registry=scenario_registry)
context.task = LinzRedistrictingContext.TASK_GS
context.scenario = 1
widget = LinzRedistrictingDockWidget(context=context, iface=IFACE)
self.assertEqual(
widget.windowTitle(),
'Redistricting - General (South Island) - Scenario 1')
context.task = LinzRedistrictingContext.TASK_GN
context.scenario = 3
widget.update_dock_title(context=context)
self.assertEqual(
widget.windowTitle(),
'Redistricting - General (North Island) - scenario 3')
def testSelection(self):
"""
Test setting/getting selected scenario
"""
layer = make_scenario_layer()
mb_electorate_layer = make_meshblock_electorate_layer()
registry = ScenarioRegistry(
source_layer=layer,
id_field='id',
name_field='name',
meshblock_electorate_layer=mb_electorate_layer)
dlg = ScenarioSelectionDialog(scenario_registry=registry)
dlg.set_selected_scenario(1)
self.assertEqual(dlg.selected_scenario(), 1)
dlg.set_selected_scenario(2)
self.assertEqual(dlg.selected_scenario(), 2)
dlg.set_selected_scenario(3)
self.assertEqual(dlg.selected_scenario(), 3)
# nothing at all selected
dlg.list.clearSelection()
self.assertIsNone(dlg.selected_scenario())
def testExportTask(self): # pylint: disable=too-many-locals, too-many-statements
"""
Test export task
"""
layer = make_scenario_layer()
mb_electorate_layer = QgsVectorLayer(
"NoGeometry?field=id:int&field=scenario_id:int&field=meshblock_number:int&field=gn_id:int&field=gs_id:int&field=m_id:int",
"source", "memory")
f = QgsFeature()
f.setAttributes([1, 1, 11, 1, 0, 7])
f2 = QgsFeature()
f2.setAttributes([2, 1, 12, 2, 0, 7])
f3 = QgsFeature()
f3.setAttributes([3, 1, 13, 2, 0, 7])
f4 = QgsFeature()
f4.setAttributes([4, 1, 14, 3, 4, 8])
f5 = QgsFeature()
f5.setAttributes([5, 1, 15, 0, 5, 8])
f6 = QgsFeature()
f6.setAttributes([6, 1, 16, 0, 5, 8])
mb_electorate_layer.dataProvider().addFeatures([f, f2, f3, f4, f5, f6])
reg = ScenarioRegistry(
source_layer=layer,
id_field='id',
name_field='name',
meshblock_electorate_layer=mb_electorate_layer
)
electorate_layer = QgsVectorLayer(
"Point?crs=EPSG:4326&field=electorate_id:int&field=code:string&field=type:string&field=estimated_pop:int&field=scenario_id:int&field=deprecated:int&field=invalid:int&field=invalid_reason:string&field=name:string&field=stats_nz_pop:int&field=stats_nz_var_20:int&field=stats_nz_var_23:int",
"source", "memory")
f = QgsFeature()
f.setAttributes([1, "test1", 'GN', 1, 0, 0, 1, 'old invalid'])
f2 = QgsFeature()
f2.setAttributes([2, "test2", 'GN', 1, 0, 0, 1, 'old invalid 2'])
f3 = QgsFeature()
f3.setAttributes([3, "test3", 'GN', 1, 0, 0, 1, 'old invalid 3'])
f4 = QgsFeature()
f4.setAttributes([4, "test4", 'GS', 1, 0, 0, 1, 'old invalid 4'])
f5 = QgsFeature()
f5.setAttributes([5, "test5", 'GS', 1, 0, 0, 1, 'old invalid 5'])
f6 = QgsFeature()
f6.setAttributes([6, "test6", 'GS', 1, 0, 0, 1, 'old invalid 6'])
f7 = QgsFeature()
f7.setAttributes([7, "test7", 'M', 1, 0, 0, 1, 'old invalid 7'])
f8 = QgsFeature()
f8.setAttributes([8, "test8", 'M', 1, 0, 0, 1, 'old invalid 8'])
electorate_layer.dataProvider().addFeatures([f, f2, f3, f4, f5, f6, f7, f8])
meshblock_layer = QgsVectorLayer(
"Polygon?crs=EPSG:4326&field=MeshblockNumber:string&field=offline_pop_m:int&field=offline_pop_gn:int&field=offline_pop_gs:int&field=staged_electorate:int&field=offshore:int",
"source", "memory")
f = QgsFeature()
f.setAttributes(["11", 5, 58900, 0])
f.setGeometry(QgsGeometry.fromWkt('Polygon((1 1, 2 1, 2 2, 1 2, 1 1))'))
f2 = QgsFeature()
f2.setAttributes(["12", 6, 57000, 0])
f2.setGeometry(QgsGeometry.fromWkt('Polygon((1 1, 2 1, 2 2, 1 2, 1 1))'))
f3 = QgsFeature()
f3.setAttributes(["13", 7, 2000, 0])
f3.setGeometry(QgsGeometry.fromWkt('Polygon((1 1, 2 1, 2 2, 1 2, 1 1))'))
f4 = QgsFeature()
f4.setAttributes(["14", 8, 0, 20])
f4.setGeometry(QgsGeometry.fromWkt('Polygon((1 1, 2 1, 2 2, 1 2, 1 1))'))
f5 = QgsFeature()
f5.setAttributes(["15", 9, 0, 30])
f5.setGeometry(QgsGeometry.fromWkt('Polygon((1 1, 2 1, 2 2, 1 2, 1 1))'))
f6 = QgsFeature()
f6.setAttributes(["16", 10, 0, 40])
f6.setGeometry(QgsGeometry.fromWkt('Polygon((1 1, 2 1, 2 2, 1 2, 1 1))'))
self.assertTrue(meshblock_layer.dataProvider().addFeatures([f, f2, f3, f4, f5, f6]))
quota_layer = make_quota_layer()
user_log_layer = make_user_log_layer()
f = QgsFeature()
f.setAttributes([1, NULL, 'user', 'v1', 1, '11', 'GN', 1, 2])
user_log_layer.dataProvider().addFeature(f)
electorate_registry = LinzElectoralDistrictRegistry(source_layer=electorate_layer, source_field='electorate_id',
title_field='code', electorate_type='GN',
quota_layer=quota_layer)
out_file = '/tmp/test.gpkg'
task = ExportTask(task_name='', dest_file=out_file, electorate_registry=electorate_registry,
meshblock_layer=meshblock_layer,
meshblock_number_field_name='MeshblockNumber', scenario_registry=reg, scenario=1,
user_log_layer=user_log_layer)
self.assertTrue(task.run(), task.message)
out_electorate_layer = QgsVectorLayer('{}|layername=electorates'.format(out_file), 'electorates', 'ogr')
self.assertTrue(out_electorate_layer.isValid())
self.assertEqual([f.attributes() for f in out_electorate_layer.getFeatures()], [[1, 'GN', 'test1', NULL],
[2, 'GN', 'test2', NULL],
[3, 'GN', 'test3', NULL],
[4, 'GS', 'test4', NULL],
[5, 'GS', 'test5', NULL],
[6, 'M', 'test7', NULL],
[7, 'M', 'test8', NULL]])
self.assertEqual([f.geometry().asWkt() for f in out_electorate_layer.getFeatures()],
['Polygon ((1 1, 2 1, 2 2, 1 2, 1 1))',
'Polygon ((2 1, 1 1, 1 2, 2 2, 2 1))',
'Polygon ((1 1, 2 1, 2 2, 1 2, 1 1))',
'Polygon ((1 1, 2 1, 2 2, 1 2, 1 1))',
'Polygon ((2 1, 1 1, 1 2, 2 2, 2 1))',
'Polygon ((2 1, 1 1, 1 2, 2 2, 2 1))',
'Polygon ((2 1, 1 1, 1 2, 2 2, 2 1))'])
out_mb_layer = QgsVectorLayer('{}|layername=meshblocks'.format(out_file), 'electorates', 'ogr')
self.assertTrue(out_mb_layer.isValid())
self.assertEqual([f.attributes() for f in out_mb_layer.getFeatures()], [[1, 11, 'test1', NULL, 'test7'],
[2, 12, 'test2', NULL, 'test7'],
[3, 13, 'test2', NULL, 'test7'],
[4, 14, 'test3', 'test4', 'test8'],
[5, 15, NULL, 'test5', 'test8'],
[6, 16, NULL, 'test5', 'test8']])
out_log_layer = QgsVectorLayer('{}|layername=user_log'.format(out_file), 'electorates', 'ogr')
self.assertTrue(out_log_layer.isValid())
self.assertEqual([f.attributes() for f in out_log_layer.getFeatures()],
[[1, 1, NULL, 'user', 'v1', 1, '11', 'GN', 1, 2]])
def testValidationTask(self): # pylint: disable=too-many-locals, too-many-statements
"""
Test validation task
"""
layer = make_scenario_layer()
mb_electorate_layer = QgsVectorLayer(
"NoGeometry?field=id:int&field=scenario_id:int&field=meshblock_number:int&field=gn_id:int&field=gs_id:int&field=m_id:int",
"source", "memory")
f = QgsFeature()
f.setAttributes([1, 1, 11, 1, 0, 7])
f2 = QgsFeature()
f2.setAttributes([2, 1, 12, 2, 0, 0])
f3 = QgsFeature()
f3.setAttributes([3, 1, 13, 2, 0, 0])
f4 = QgsFeature()
f4.setAttributes([4, 1, 14, 3, 4, 0])
f5 = QgsFeature()
f5.setAttributes([5, 1, 15, 3, 5, 0])
f6 = QgsFeature()
f6.setAttributes([6, 1, 16, 2, 5, 0])
mb_electorate_layer.dataProvider().addFeatures([f, f2, f3, f4, f5, f6])
reg = ScenarioRegistry(source_layer=layer,
id_field='id',
name_field='name',
meshblock_electorate_layer=mb_electorate_layer)
electorate_layer = QgsVectorLayer(
"Point?crs=EPSG:4326&field=electorate_id:int&field=code:string&field=type:string&field=estimated_pop:int&field=scenario_id:int&field=deprecated:int&field=invalid:int&field=invalid_reason:string&field=name:string&field=stats_nz_pop:int&field=stats_nz_var_20:int&field=stats_nz_var_23:int&field=electorate_id_stats:string&field=expected_regions:int",
"source", "memory")
f = QgsFeature()
f.setAttributes([
1, "test1", 'GN', 1, 0, 0, 1, 'old invalid', NULL, NULL, NULL,
NULL, NULL, 1
])
f2 = QgsFeature()
f2.setAttributes([
2, "test2", 'GN', 1, 0, 0, 1, 'old invalid 2', NULL, NULL, NULL,
NULL, NULL, 2
])
f3 = QgsFeature()
f3.setAttributes([
3, "test3", 'GN', 1, 0, 0, 1, 'old invalid 3', NULL, NULL, NULL,
NULL, NULL, 2
])
f4 = QgsFeature()
f4.setAttributes([
4, "test4", 'GS', 1, 0, 0, 1, 'old invalid 4', NULL, NULL, NULL,
NULL, NULL, 2
])
f5 = QgsFeature()
f5.setAttributes([
5, "test5", 'GS', 1, 0, 0, 1, 'old invalid 5', NULL, NULL, NULL,
NULL, NULL, 1
])
f6 = QgsFeature()
f6.setAttributes([
6, "test6", 'GS', 1, 0, 0, 1, 'old invalid 6', NULL, NULL, NULL,
NULL, NULL, 1
])
f7 = QgsFeature()
f7.setAttributes([
7, "test7", 'M', 1, 0, 1, 1, 'old invalid 7', NULL, NULL, NULL,
NULL, NULL, 1
])
f8 = QgsFeature()
# deprecated
f8.setAttributes([
8, "test8", 'M', 1, 0, 1, 1, 'old invalid 8', NULL, NULL, NULL,
NULL, NULL, 1
])
electorate_layer.dataProvider().addFeatures(
[f, f2, f3, f4, f5, f6, f7, f8])
meshblock_layer = QgsVectorLayer(
"Point?crs=EPSG:4326&field=MeshblockNumber:string&field=offline_pop_m:int&field=offline_pop_gn:int&field=offline_pop_gs:int&field=staged_electorate:int&field=offshore:int",
"source", "memory")
f = QgsFeature()
f.setAttributes(["11", 5, 58900, 0, NULL, 0])
f.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(1, 2)))
f2 = QgsFeature()
f2.setAttributes(["12", 6, 57000, 0, NULL, 0])
f2.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(2, 3)))
f3 = QgsFeature()
f3.setAttributes(["13", 7, 2000, 0, NULL, 0])
f3.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(4, 5)))
f4 = QgsFeature()
f4.setAttributes(["14", 8, 0, 20, NULL, 0])
f4.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(6, 7)))
f5 = QgsFeature()
f5.setAttributes(["15", 9, 0, 30, NULL, 0])
f5.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(8, 9)))
f6 = QgsFeature()
f6.setAttributes(["16", 10, 0, 40, NULL, 1])
f6.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(10, 11)))
meshblock_layer.dataProvider().addFeatures([f, f2, f3, f4, f5, f6])
quota_layer = make_quota_layer()
electorate_registry = LinzElectoralDistrictRegistry(
source_layer=electorate_layer,
source_field='electorate_id',
title_field='code',
electorate_type='GN',
quota_layer=quota_layer)
task = ValidationTask(task_name='',
electorate_registry=electorate_registry,
meshblock_layer=meshblock_layer,
meshblock_number_field_name='MeshblockNumber',
scenario_registry=reg,
scenario=1,
task='GN')
self.assertEqual(
[f.attributes()[:9] for f in electorate_layer.getFeatures()],
[[1, 'test1', 'GN', 1, 0, 0, NULL, NULL, NULL],
[2, 'test2', 'GN', 1, 0, 0, NULL, NULL, NULL],
[3, 'test3', 'GN', 1, 0, 0, NULL, NULL, NULL],
[4, 'test4', 'GS', 1, 0, 0, 1, 'old invalid 4', NULL],
[5, 'test5', 'GS', 1, 0, 0, 1, 'old invalid 5', NULL],
[6, 'test6', 'GS', 1, 0, 0, 1, 'old invalid 6', NULL],
[7, 'test7', 'M', 1, 0, 1, 1, 'old invalid 7', NULL],
[8, 'test8', 'M', 1, 0, 1, 1, 'old invalid 8', NULL]])
self.assertTrue(task.run())
self.assertEqual(len(task.results), 5)
self.assertEqual(task.results[0][ValidationTask.ELECTORATE_ID], 2)
self.assertEqual(task.results[0][ValidationTask.ELECTORATE_NAME],
'test2')
self.assertEqual(task.results[0][ValidationTask.ERROR],
'Electorate is non-contiguous')
self.assertEqual(
task.results[0][ValidationTask.ELECTORATE_GEOMETRY].asWkt(0),
'MultiPoint ((2 3),(4 5),(10 11))')
self.assertEqual(task.results[1][ValidationTask.ELECTORATE_ID], 2)
self.assertEqual(task.results[1][ValidationTask.ELECTORATE_NAME],
'test2')
self.assertEqual(task.results[1][ValidationTask.ERROR],
'Contiguous part 1')
self.assertEqual(
task.results[1][ValidationTask.ELECTORATE_GEOMETRY].asWkt(0),
'Point (2 3)')
self.assertEqual(task.results[2][ValidationTask.ELECTORATE_ID], 2)
self.assertEqual(task.results[2][ValidationTask.ELECTORATE_NAME],
'test2')
self.assertEqual(task.results[2][ValidationTask.ERROR],
'Contiguous part 2')
self.assertEqual(
task.results[2][ValidationTask.ELECTORATE_GEOMETRY].asWkt(0),
'Point (4 5)')
self.assertEqual(task.results[3][ValidationTask.ELECTORATE_ID], 2)
self.assertEqual(task.results[3][ValidationTask.ELECTORATE_NAME],
'test2')
self.assertEqual(task.results[3][ValidationTask.ERROR],
'Contiguous part 3')
self.assertEqual(
task.results[3][ValidationTask.ELECTORATE_GEOMETRY].asWkt(0),
'Point (10 11)')
self.assertEqual(task.results[4][ValidationTask.ELECTORATE_ID], 3)
self.assertEqual(task.results[4][ValidationTask.ELECTORATE_NAME],
'test3')
self.assertEqual(task.results[4][ValidationTask.ERROR],
'Outside quota tolerance')
self.assertEqual(
[f.attributes()[:9] for f in electorate_layer.getFeatures()],
[[1, 'test1', 'GN', 58900, 1, 0, 0, NULL, NULL],
[
2, 'test2', 'GN', 1, 0, 0, 1, 'Electorate is non-contiguous',
NULL
], [
3, 'test3', 'GN', 1, 0, 0, 1, 'Outside quota tolerance', NULL
], [4, 'test4', 'GS', 1, 0, 0, 1, 'old invalid 4', NULL],
[5, 'test5', 'GS', 1, 0, 0, 1, 'old invalid 5', NULL],
[6, 'test6', 'GS', 1, 0, 0, 1, 'old invalid 6', NULL],
[7, 'test7', 'M', 1, 0, 1, 1, 'old invalid 7', NULL],
[8, 'test8', 'M', 1, 0, 1, 1, 'old invalid 8', NULL]])
electorate_registry = LinzElectoralDistrictRegistry(
source_layer=electorate_layer,
source_field='electorate_id',
title_field='code',
electorate_type='GN',
quota_layer=quota_layer)
task = ValidationTask(task_name='',
electorate_registry=electorate_registry,
meshblock_layer=meshblock_layer,
meshblock_number_field_name='MeshblockNumber',
scenario_registry=reg,
scenario=1,
task='GS')
self.assertTrue(task.run())
self.assertEqual(len(task.results), 7)
self.assertEqual(task.results[0][ValidationTask.ELECTORATE_ID], 4)
self.assertEqual(task.results[0][ValidationTask.ELECTORATE_NAME],
'test4')
self.assertEqual(task.results[0][ValidationTask.ERROR],
'Outside quota tolerance')
self.assertEqual(task.results[1][ValidationTask.ELECTORATE_ID], 4)
self.assertEqual(task.results[1][ValidationTask.ELECTORATE_NAME],
'test4')
self.assertEqual(task.results[1][ValidationTask.ERROR],
'Electorate has less parts than expected')
self.assertEqual(task.results[2][ValidationTask.ELECTORATE_ID], 5)
self.assertEqual(task.results[2][ValidationTask.ELECTORATE_NAME],
'test5')
self.assertEqual(task.results[2][ValidationTask.ERROR],
'Outside quota tolerance')
self.assertEqual(task.results[3][ValidationTask.ELECTORATE_ID], 5)
self.assertEqual(task.results[3][ValidationTask.ELECTORATE_NAME],
'test5')
self.assertEqual(task.results[3][ValidationTask.ERROR],
'Electorate is non-contiguous')
self.assertEqual(
task.results[3][ValidationTask.ELECTORATE_GEOMETRY].asWkt(0),
'MultiPoint ((8 9),(10 11))')
self.assertEqual(task.results[4][ValidationTask.ELECTORATE_ID], 5)
self.assertEqual(task.results[4][ValidationTask.ELECTORATE_NAME],
'test5')
self.assertEqual(task.results[4][ValidationTask.ERROR],
'Contiguous part 1')
self.assertEqual(
task.results[4][ValidationTask.ELECTORATE_GEOMETRY].asWkt(0),
'Point (8 9)')
self.assertEqual(task.results[5][ValidationTask.ELECTORATE_ID], 5)
self.assertEqual(task.results[5][ValidationTask.ELECTORATE_NAME],
'test5')
self.assertEqual(task.results[5][ValidationTask.ERROR],
'Contiguous part 2')
self.assertEqual(
task.results[5][ValidationTask.ELECTORATE_GEOMETRY].asWkt(0),
'Point (10 11)')
self.assertEqual(task.results[6][ValidationTask.ELECTORATE_ID], 6)
self.assertEqual(task.results[6][ValidationTask.ELECTORATE_NAME],
'test6')
self.assertEqual(task.results[6][ValidationTask.ERROR],
'Outside quota tolerance')
self.assertEqual([
f.attributes()[:9] for f in electorate_layer.getFeatures()
], [[1, 'test1', 'GN', 58900, 1, 0, 0, NULL, NULL],
[
2, 'test2', 'GN', 1, 0, 0, 1, 'Electorate is non-contiguous',
NULL
], [3, 'test3', 'GN', 1, 0, 0, 1, 'Outside quota tolerance', NULL],
[
4, 'test4', 'GS', 1, 0, 0, 1,
'Electorate has less parts than expected', NULL
],
[
5, 'test5', 'GS', 1, 0, 0, 1, 'Electorate is non-contiguous',
NULL
], [6, 'test6', 'GS', 1, 0, 0, 1, 'Outside quota tolerance', NULL],
[7, 'test7', 'M', 1, 0, 1, 1, 'old invalid 7', NULL],
[8, 'test8', 'M', 1, 0, 1, 1, 'old invalid 8', NULL]])
electorate_registry = LinzElectoralDistrictRegistry(
source_layer=electorate_layer,
source_field='electorate_id',
title_field='code',
electorate_type='M',
quota_layer=quota_layer)
task = ValidationTask(task_name='',
electorate_registry=electorate_registry,
meshblock_layer=meshblock_layer,
meshblock_number_field_name='MeshblockNumber',
scenario_registry=reg,
scenario=1,
task='M')
self.assertTrue(task.run())
self.assertEqual(len(task.results), 1)
self.assertEqual(task.results[0][ValidationTask.ELECTORATE_ID], 7)
self.assertEqual(task.results[0][ValidationTask.ELECTORATE_NAME],
'test7')
self.assertEqual(task.results[0][ValidationTask.ERROR],
'Deprecated electorate has meshblocks assigned')
self.assertEqual([
f.attributes()[:9] for f in electorate_layer.getFeatures()
], [[1, 'test1', 'GN', 58900, 1, 0, 0, NULL, NULL],
[
2, 'test2', 'GN', 1, 0, 0, 1, 'Electorate is non-contiguous',
NULL
], [3, 'test3', 'GN', 1, 0, 0, 1, 'Outside quota tolerance', NULL],
[
4, 'test4', 'GS', 1, 0, 0, 1,
'Electorate has less parts than expected', NULL
],
[
5, 'test5', 'GS', 1, 0, 0, 1, 'Electorate is non-contiguous',
NULL
], [6, 'test6', 'GS', 1, 0, 0, 1, 'Outside quota tolerance', NULL],
[
7, 'test7', 'M', 1, 0, 1, 1,
'Deprecated electorate has meshblocks assigned', NULL
], [8, 'test8', 'M', 0, 1, 1, 0, NULL, NULL]])
