def test_parse_input(self):
function_input = {
'thresholds': lambda: [1.0],
'postprocessors': {
'Gender': {'on': lambda: True},
'Age': {
'on': lambda: True,
'params': {
'youth_ratio': lambda: 0.263,
'elderly_ratio': lambda: 0.078,
'adult_ratio': lambda: 0.659}}}}
dialog = FunctionOptionsDialog()
result = dialog.parse_input(function_input)
print result
expected = OrderedDict([
('thresholds', [1.0]),
('postprocessors', OrderedDict([
('Gender', OrderedDict([('on', True)])),
('Age', OrderedDict([
('on', True),
('params', OrderedDict([
('elderly_ratio', 0.078),
('youth_ratio', 0.263),
('adult_ratio', 0.659)]))]))]))])
# noinspection PyPep8Naming
self.maxDiff = None
self.assertDictEqual(result, expected)
def test_build_form_minimum_needs(self):
"""Test that we can build a form by passing it a function and params.
"""
function_id = 'Flood Evacuation Function Vector Hazard'
function_list = get_plugins(function_id)
assert len(function_list) == 1
assert function_list[0].keys()[0] == function_id
dialog = FunctionOptionsDialog()
parameters = {
'thresholds': [1.0],
'postprocessors': {
'Gender': {'on': True},
'Age': {
'on': True,
'params': {
'youth_ratio': 0.263,
'elderly_ratio': 0.078,
'adult_ratio': 0.659}}}}
dialog.build_form(parameters)
assert dialog.tabWidget.count() == 2
children = dialog.tabWidget.findChildren(QLineEdit)
assert len(children) == 4
def test_build_form(self):
"""Test that we can build a form by passing it a function and params.
"""
function_id = 'ITB Fatality Function'
function_list = get_plugins(function_id)
assert len(function_list) == 1
assert function_list[0].keys()[0] == function_id
dialog = FunctionOptionsDialog()
# Define rice for minimum needs
rice = ResourceParameter()
rice.value = 2.8
rice.frequency = 'weekly'
rice.minimum_allowed_value = 1.4
rice.maximum_allowed_value = 5.6
rice.name = 'Rice'
rice.unit.abbreviation = 'kg'
rice.unit.name = 'kilogram'
rice.unit.plural = 'kilograms'
parameter = {
'thresholds': [1.0],
'postprocessors': {
'Gender': {'on': True},
'Age': {
'on': True,
'params': {
'youth_ratio': 0.263,
'elderly_ratio': 0.078,
'adult_ratio': 0.659
}
}
},
'minimum needs': [rice]
}
dialog.build_form(parameter)
message = 'There should be %s tabwidget but got %s' % (
3, dialog.tabWidget.count())
self.assertEqual(dialog.tabWidget.count(), 3, message)
children = dialog.tabWidget.findChildren(QLineEdit)
message = 'There should be %s QLineEdit but got %s' % (
5, len(children))
self.assertEqual(len(children), 5, message)
def test_build_widget(self):
dialog = FunctionOptionsDialog()
value = dialog.build_widget(dialog.configLayout, 'foo', [2.3])
widget = dialog.findChild(QLineEdit)
# initial value must be same with default
expected_value = [2.3]
real_value = value()
message = 'Expected %s but got %s' % (expected_value, real_value)
self.assertEqual(expected_value, real_value, message)
# change to 5.9
widget.setText('5.9')
expected_value = [5.9]
real_value = value()
message = 'Expected %s but got %s' % (expected_value, real_value)
self.assertEqual(expected_value, real_value, message)
widget.setText('5.9, 70')
expected_value = [5.9, 70]
real_value = value()
message = 'Expected %s but got %s' % (expected_value, real_value)
self.assertEqual(expected_value, real_value, message)
widget.setText('bar')
try:
value()
except ValueError:
# expected to raises this exception
pass
else:
raise Exception("Fail: must be raise an exception")
dialog = FunctionOptionsDialog()
value = dialog.build_widget(dialog.configLayout, 'foo', True)
widget = dialog.findChild(QCheckBox)
# initial value must be same with default
expected_value = True
real_value = value()
message = 'Expected %s but got %s' % (expected_value, real_value)
self.assertEqual(expected_value, real_value, message)
widget.setChecked(False)
expected_value = False
real_value = value()
message = 'Expected %s but got %s' % (expected_value, real_value)
self.assertEqual(expected_value, real_value, message)
original_value = {'a': 1, 'b': 2}
dialog = FunctionOptionsDialog()
value = dialog.build_widget(dialog.configLayout, 'foo', original_value)
widget = dialog.findChild(QLineEdit)
# initial value must be same with default
expected_value = original_value
real_value = value()
message = 'Expected %s but got %s' % (expected_value, real_value)
self.assertEqual(expected_value, real_value, message)
expected_value = {'a': 2, 'b': 1}
widget.setText(str(expected_value))
real_value = value()
message = 'Expected %s but got %s' % (expected_value, real_value)
self.assertEqual(expected_value, real_value, message)
