def testClickHandler(self):
"""
Tests the request population callback
"""
layer = QgsVectorLayer(
"Point?crs=EPSG:4326&field=fld1:string&field=fld2:string&field=type:string&field=estimated_pop:int&field=deprecated:int&field=stats_nz_pop:int&field=stats_nz_var_20:int&field=stats_nz_var_23:int&field=scenario_id:int",
"source", "memory")
f = QgsFeature()
f.setAttributes(["test4", "xtest1", 'GN', 1000])
layer.dataProvider().addFeatures([f])
quota_layer = make_quota_layer()
registry = LinzElectoralDistrictRegistry(
source_layer=layer,
quota_layer=quota_layer,
electorate_type='',
source_field='fld1',
title_field='fld2')
dock = RedistrictingDockWidget(IFACE)
handler = LinzRedistrictGuiHandler(redistrict_dock=dock, district_registry=registry)
handler.request_population_callback = None
handler.show_stats_for_district('test4')
handler.request_population()
def callback(electorate):
"""
Dummy callback
"""
callback.electorate = electorate
callback.electorate = None
handler.request_population_callback = callback
handler.request_population()
self.assertEqual(callback.electorate, 'test4')
def testDialog(self):
"""
Test dialog functionality
"""
layer = QgsVectorLayer(
"Point?crs=EPSG:4326&field=fld1:string&field=code:string&field=type:string&field=estimated_pop:int&field=deprecated:int&field=stats_nz_pop:int&field=stats_nz_var_20:int&field=stats_nz_var_23:int&field=scenario_id:int&field=electorate_id_stats:string&field=expected_regions:int",
"source", "memory")
f = QgsFeature()
f.setAttributes(["test4", "xtest1", 'GN', 1, False])
f2 = QgsFeature()
f2.setAttributes(["test2", "xtest3", 'GN', 2, True])
f3 = QgsFeature()
f3.setAttributes(["test3", "xtest3", 'GN', 1, False])
layer.dataProvider().addFeatures([f, f2, f3])
quota_layer = make_quota_layer()
reg = LinzElectoralDistrictRegistry(source_layer=layer,
quota_layer=quota_layer,
electorate_type='',
source_field='fld1',
title_field='fld1')
dlg = DeprecateElectorateDialog(electorate_registry=reg)
self.assertIsNotNone(dlg)
self.assertEqual(
[dlg.list.item(r).text() for r in range(dlg.list.count())],
['test4', '*test2', 'test3'])
def testShowStats(self): # pylint: disable=too-many-statements
"""Test show stats for district"""
layer = QgsVectorLayer(
"Point?crs=EPSG:4326&field=fld1:string&field=fld2:string&field=type:string&field=estimated_pop:int&field=deprecated:int&field=stats_nz_pop:int&field=stats_nz_var_20:int&field=stats_nz_var_23:int&field=scenario_id:int",
"source", "memory")
f = QgsFeature()
f.setAttributes(["test4", "xtest1", 'GN', 1000])
f2 = QgsFeature()
f2.setAttributes(["test2", "xtest3", 'GS', 2000])
f3 = QgsFeature()
f3.setAttributes(["test3", "xtest3", 'M', 63000])
layer.dataProvider().addFeatures([f, f2, f3])
quota_layer = make_quota_layer()
registry = LinzElectoralDistrictRegistry(
source_layer=layer,
quota_layer=quota_layer,
electorate_type='',
source_field='fld1',
title_field='fld2')
dock = RedistrictingDockWidget(IFACE)
handler = LinzRedistrictGuiHandler(redistrict_dock=dock, district_registry=registry)
handler.show_stats_for_district('')
self.assertFalse(dock.frame.toPlainText())
handler.show_stats_for_district('test4')
self.assertIn('Statistics for xtest1', dock.frame.toPlainText())
self.assertIn('General North Island', dock.frame.toPlainText())
self.assertNotIn('updating', dock.frame.toPlainText())
self.assertIn('59000', dock.frame.toPlainText())
self.assertIn('1000*', dock.frame.toPlainText())
self.assertIn('-98%', dock.frame.toPlainText())
self.assertIn('color:#ff0000', dock.frame.toHtml())
self.assertIn('estimated population available', dock.frame.toPlainText())
handler.show_stats_for_district('test3')
self.assertIn('Statistics for xtest3', dock.frame.toPlainText())
self.assertIn('Māori', dock.frame.toPlainText())
self.assertNotIn('updating', dock.frame.toPlainText())
self.assertIn('61000', dock.frame.toPlainText())
self.assertIn('63000*', dock.frame.toPlainText())
self.assertIn('+3%', dock.frame.toPlainText())
self.assertNotIn('color:#ff0000', dock.frame.toHtml())
self.assertIn('estimated population available', dock.frame.toPlainText())
registry.flag_stats_nz_updating('test4')
handler.show_stats_for_district('test4')
self.assertIn('Statistics for xtest1', dock.frame.toPlainText())
self.assertIn('General North Island', dock.frame.toPlainText())
self.assertIn('updating', dock.frame.toPlainText())
self.assertNotIn('1000*', dock.frame.toPlainText())
self.assertNotIn('-98%', dock.frame.toPlainText())
self.assertNotIn('color:#ff0000', dock.frame.toHtml())
self.assertNotIn('estimated population available', dock.frame.toPlainText())
registry.update_stats_nz_values('test3', {'currentPopulation': 1111,
'varianceYear1': 1.5,
'varianceYear2': -1.1})
handler.show_stats_for_district('test3')
self.assertIn('Statistics for xtest3', dock.frame.toPlainText())
self.assertIn('Māori', dock.frame.toPlainText())
self.assertNotIn('updating', dock.frame.toPlainText())
self.assertIn('61000', dock.frame.toPlainText())
self.assertIn('1111', dock.frame.toPlainText())
self.assertIn('1.5', dock.frame.toPlainText())
self.assertIn('-1.1', dock.frame.toPlainText())
self.assertIn('-98%', dock.frame.toPlainText())
self.assertIn('color:#ff0000', dock.frame.toHtml())
self.assertNotIn('estimated population available', dock.frame.toPlainText())
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]])
def testDialog(self):
"""
Test dialog functionality
"""
scenario_layer = make_scenario_layer()
scenario_registry = ScenarioRegistry(source_layer=scenario_layer,
id_field='id',
name_field='name',
meshblock_electorate_layer=None)
layer = QgsVectorLayer(
"Point?crs=EPSG:4326&field=fld1:string&field=code:string&field=type:string&field=estimated_pop:int&field=deprecated:int&field=stats_nz_pop:int&field=stats_nz_var_20:int&field=stats_nz_var_23:int&field=scenario_id:int",
"source", "memory")
f = QgsFeature()
f.setAttributes(["test4", "xtest1", 'GN'])
f2 = QgsFeature()
f2.setAttributes(["test2", "xtest3", 'GS'])
f3 = QgsFeature()
f3.setAttributes(["test3", "xtest3", 'M'])
layer.dataProvider().addFeatures([f, f2, f3])
quota_layer = make_quota_layer()
reg = LinzElectoralDistrictRegistry(source_layer=layer,
quota_layer=quota_layer,
electorate_type='',
source_field='fld1',
title_field='fld1')
context = LinzRedistrictingContext(scenario_registry=scenario_registry)
dlg = CreateElectorateDialog(registry=reg, context=context)
self.assertIsNotNone(dlg)
self.assertFalse(
dlg.button_box.button(QDialogButtonBox.Ok).isEnabled())
dlg.name_line_edit.setText('new district')
dlg.code_line_edit.setText('new code')
self.assertEqual(dlg.name(), 'new district')
self.assertEqual(dlg.code(), 'new code')
# dupe name
dlg.name_line_edit.setText('test4')
self.assertFalse(
dlg.button_box.button(QDialogButtonBox.Ok).isEnabled())
self.assertIn('already exists', dlg.feedback_label.text())
dlg.name_line_edit.setText('test99')
self.assertTrue(dlg.button_box.button(QDialogButtonBox.Ok).isEnabled())
self.assertFalse(dlg.feedback_label.text())
dlg.name_line_edit.setText('')
self.assertFalse(
dlg.button_box.button(QDialogButtonBox.Ok).isEnabled())
self.assertIn('must be entered', dlg.feedback_label.text())
dlg.name_line_edit.setText('test99')
self.assertTrue(dlg.button_box.button(QDialogButtonBox.Ok).isEnabled())
self.assertFalse(dlg.feedback_label.text())
# dupe code
dlg.code_line_edit.setText('xtest1')
self.assertFalse(
dlg.button_box.button(QDialogButtonBox.Ok).isEnabled())
self.assertIn('already exists', dlg.feedback_label.text())
dlg.code_line_edit.setText('test99')
self.assertTrue(dlg.button_box.button(QDialogButtonBox.Ok).isEnabled())
self.assertFalse(dlg.feedback_label.text())
dlg.code_line_edit.setText('')
self.assertIn('must be entered', dlg.feedback_label.text())
self.assertFalse(
dlg.button_box.button(QDialogButtonBox.Ok).isEnabled())
dlg.code_line_edit.setText('test99')
self.assertFalse(dlg.feedback_label.text())
