def test_minimum_needs_calculator(self):
"""Test behaviour of the minimum needs function."""
dialog = NeedsCalculatorDialog(PARENT)
layer = load_test_vector_layer('other', 'minimum_needs.shp')
QgsProject.instance().addMapLayers([layer])
# Set selected layer and displaced field
dialog.layer.setLayer(layer)
dialog.displaced.setField('displaced')
# run minimum needs function
dialog.accept()
# get output layer
layer = dialog.result_layer
assert layer is not None
field_names = [field.name() for field in layer.fields()]
for feature in layer.getFeatures():
value = [attribute for attribute in feature.attributes()]
actual_attributes = dict(list(zip(field_names, value)))
expected_attributes = {
'displaced': 1000,
'minimum_needs__rice': 2800,
'minimum_needs__drinking_water': 17500,
'minimum_needs__clean_water': 67000,
'minimum_needs__family_kits': 200,
'minimum_needs__toilets': 50
}
self.assertDictEqual(byteify(expected_attributes),
byteify(actual_attributes))
def test_minimum_needs_calculator(self):
"""Test behaviour of the minimum needs function."""
dialog = NeedsCalculatorDialog(PARENT)
layer = load_test_vector_layer('other', 'minimum_needs.shp')
QgsMapLayerRegistry.instance().addMapLayers([layer])
# Set selected layer and displaced field
dialog.layer.setLayer(layer)
dialog.displaced.setField(u'displaced')
# run minimum needs function
dialog.accept()
# get output layer
layer = dialog.result_layer
assert layer is not None
field_names = [field.name() for field in layer.pendingFields()]
for feature in layer.getFeatures():
value = [attribute for attribute in feature.attributes()]
actual_attributes = dict(zip(field_names, value))
expected_attributes = {
'displaced': 1000,
'minimum_needs__rice': 2800,
'minimum_needs__drinking_water': 17500,
'minimum_needs__clean_water': 67000,
'minimum_needs__family_kits': 200,
'minimum_needs__toilets': 50}
self.assertDictEqual(
byteify(expected_attributes),
byteify(actual_attributes))
def test_minimum_needs_calculator(self):
"""Test behaviour of the minimum needs function."""
dialog = NeedsCalculatorDialog(PARENT)
layer = QgsVectorLayer(shapefile_path,
os.path.basename(shapefile_path), 'ogr')
QgsMapLayerRegistry.instance().addMapLayers([layer])
# Set selected layer and displaced field
dialog.layer.setLayer(layer)
dialog.displaced.setField(u'displaced')
# run minimum needs function
dialog.accept()
# get output layer
layer = dialog.result_layer
assert layer is not None
field_names = [field.name() for field in layer.pendingFields()]
for feature in layer.getFeatures():
value = [attribute for attribute in feature.attributes()]
actual_attributes = dict(zip(field_names, value))
expected_attributes = {
'displaced': 1000,
'minimum_needs__rice': 2800,
'minimum_needs__drinking_water': 17500,
'minimum_needs__clean_water': 67000,
'minimum_needs__family_kits': 200,
'minimum_needs__toilets': 50
}
self.assertDictEqual(expected_attributes, actual_attributes)
