def show_keywords_wizard(self): """Show the keywords creation wizard.""" # import here only so that it is AFTER i18n set up from safe_qgis.tools.wizard_dialog import WizardDialog if self.iface.activeLayer() is None: return dialog = WizardDialog(self.iface.mainWindow(), self.iface, self.dock_widget) dialog.exec_() # modal
def show_keywords_wizard(self): """Show the keywords creation wizard.""" # import here only so that it is AFTER i18n set up from safe_qgis.tools.wizard_dialog import WizardDialog if self.iface.activeLayer() is None: return dialog = WizardDialog( self.iface.mainWindow(), self.iface, self.dock_widget) dialog.exec_() # modal
def test_existing_aggregation_keywords(self): """Test for case existing keywords in aggregation layer.""" layer = clone_shp_layer( name='kabupaten_jakarta', include_keywords=True, directory=BOUNDDATA) dialog = WizardDialog(PARENT, IFACE, None, layer) category = dialog.lstCategories.currentItem().text() expected_category = 'aggregation' message = 'Expected %s but I got %s.' % (expected_category, category) self.assertEqual(expected_category, category, message) dialog.pbnNext.click() expected_aggregation_attribute = 'KAB_NAME' aggregation_attribute = dialog.lstFields.currentItem().text() message = 'Expected %s but I got %s.' % ( expected_aggregation_attribute, aggregation_attribute) self.assertEqual( expected_aggregation_attribute, aggregation_attribute, message) dialog.pbnNext.click() expected_aggregation_attributes = { 'elderly ratio attribute': 'Use default', 'youth ratio default': 0.26, 'elderly ratio default': 0.08, 'adult ratio attribute': 'Use default', 'female ratio attribute': 'Use default', 'youth ratio attribute': 'Use default', 'female ratio default': 0.5, 'adult ratio default': 0.66 } aggregation_attributes = dialog.get_aggregation_attributes() message = 'Expected %s but I got %s.' % ( expected_aggregation_attributes, aggregation_attributes) self.assertDictEqual( expected_aggregation_attributes, aggregation_attributes, message) dialog.cboFemaleRatioAttribute.setCurrentIndex(2) expected_female_attribute_key = 'PEREMPUAN' female_attribute_key = dialog.cboFemaleRatioAttribute.currentText() message = 'Expected %s but I got %s.' % ( expected_female_attribute_key, female_attribute_key) self.assertEqual( expected_female_attribute_key, female_attribute_key, message) is_enabled = dialog.dsbFemaleRatioDefault.isEnabled() message = 'Expected disabled but I got enabled.' self.assertEqual(is_enabled, False, message) remove_temp_file(layer.source())
def test_integrated_point(self): """Test for point layer and all possibilities.""" layer = clone_shp_layer(name='Marapi', source_directory=HAZDATA) dialog = WizardDialog(layer=layer) expected_categories = ['hazard', 'exposure'] self.check_list(expected_categories, dialog.lstCategories) self.select_from_list_widget('hazard', dialog.lstCategories) self.check_current_text('hazard', dialog.lstCategories) dialog.pbnNext.click() # go to subcategory # check if in step subcategory self.check_current_step(step_subcategory, dialog) expected_subcategories = ['volcano'] self.check_list(expected_subcategories, dialog.lstSubcategories) self.check_current_text('volcano', dialog.lstSubcategories) dialog.pbnNext.click() # go to source # check if in step source self.check_current_step(step_source, dialog) dialog.pbnNext.click() # go to title # check if in step title self.check_current_step(step_title, dialog) dialog.pbnCancel.click()
def test_translation(self): """Test for metadata translation.""" from safe_qgis.tools.wizard_dialog import WizardDialog from safe_qgis.utilities.utilities_for_testing import ( clone_shp_layer, remove_vector_temp_file) from safe_qgis.safe_interface import BOUNDDATA from safe.common.testing import get_qgis_app # Get QGis app handle # noinspection PyPep8Naming _, _, IFACE, PARENT = get_qgis_app() layer = clone_shp_layer( name='kabupaten_jakarta', include_keywords=True, source_directory=BOUNDDATA) # noinspection PyTypeChecker dialog = WizardDialog(PARENT, IFACE, None, layer) expected_categories = ['keterpaparan', 'ancaman', 'agregasi'] # noinspection PyTypeChecker self.check_list(expected_categories, dialog.lstCategories) self.check_current_text('agregasi', dialog.lstCategories) dialog.pbnNext.click() remove_vector_temp_file(layer.source())
def test_unknown_unit(self): """Checking that it works for unknown unit.""" layer = clone_shp_layer(name='Marapi_evac_zone_3000m', include_keywords=True, source_directory=HAZDATA) dialog = WizardDialog(layer=layer) dialog.pbnNext.click() # choose hazard go to subcategory step dialog.pbnNext.click() # choose volcano go to unit step dialog.lstUnits.setCurrentRow(0) # Choose volcano categorical self.check_current_text('volcano categorical', dialog.lstUnits) dialog.pbnNext.click() # choose volcano go to field step dialog.lstFields.setCurrentRow(0) # Choose Radius self.check_current_text('Radius', dialog.lstFields) dialog.pbnNext.click() # choose volcano go to classify step # check if in step classify self.check_current_step(step_classify, dialog) dialog.pbnNext.click() # choose volcano go to source step # check if in step source self.check_current_step(step_source, dialog) dialog.pbnNext.click() # choose volcano go to title step # check if in step title self.check_current_step(step_title, dialog)
def test_sum_ratio_behavior(self): """Test for wizard's behavior related sum of age ratio.""" layer = clone_shp_layer( name='kabupaten_jakarta', include_keywords=True, source_directory=BOUNDDATA) dialog = WizardDialog(layer=layer) dialog.suppress_warning_dialog = True self.check_current_text('aggregation', dialog.lstCategories) dialog.pbnNext.click() # Go to unit step self.check_current_text('KAB_NAME', dialog.lstFields) dialog.pbnNext.click() # Go to aggregation step dialog.dsbYouthRatioDefault.setValue(1.0) dialog.pbnNext.click() # Try to go to source step # check if still in aggregation step self.check_current_step(step_aggregation, dialog) dialog.cboYouthRatioAttribute.setCurrentIndex(1) # set don't use dialog.pbnNext.click() # Try to go to source step # check if in source step self.check_current_step(step_source, dialog) dialog.pbnBack.click() # Go to aggregation step # check if in aggregation step self.check_current_step(step_aggregation, dialog) dialog.cboYouthRatioAttribute.setCurrentIndex(0) # set global default dialog.dsbYouthRatioDefault.setValue(0.0) dialog.pbnNext.click() # Try to go to source step # check if in source step self.check_current_step(step_source, dialog) dialog.pbnCancel.click()
def test_point_layer(self): """Wizard for point layer.""" layer = clone_shp_layer(name='Marapi', include_keywords=True, source_directory=HAZDATA) dialog = WizardDialog(layer=layer) dialog.pbnNext.click() # choose hazard go to subcategory step dialog.pbnNext.click() # choose volcano go to source step # check if in step source self.check_current_step(step_source, dialog) dialog.pbnNext.click() # choose volcano go to title step # check if in step title self.check_current_step(step_title, dialog) dialog.accept()
def test_existing_aggregation_keywords(self): """Test for case existing keywords in aggregation layer.""" layer = clone_shp_layer(name='kabupaten_jakarta', include_keywords=True, source_directory=BOUNDDATA) dialog = WizardDialog(layer=layer) category = dialog.lstCategories.currentItem().text() expected_category = 'aggregation' message = 'Expected %s but I got %s.' % (expected_category, category) self.assertEqual(expected_category, category, message) dialog.pbnNext.click() self.check_current_text('KAB_NAME', dialog.lstFields) dialog.pbnNext.click() expected_aggregation_attributes = { 'elderly ratio attribute': 'Global default', 'youth ratio default': 0.26, 'elderly ratio default': 0.08, 'adult ratio attribute': 'Global default', 'female ratio attribute': 'Global default', 'youth ratio attribute': 'Global default', 'female ratio default': 0.5, 'adult ratio default': 0.66 } aggregation_attributes = dialog.get_aggregation_attributes() message = 'Expected %s but I got %s.' % ( expected_aggregation_attributes, aggregation_attributes) self.assertDictEqual(expected_aggregation_attributes, aggregation_attributes, message) dialog.cboFemaleRatioAttribute.setCurrentIndex(2) expected_female_attribute_key = 'PEREMPUAN' female_attribute_key = dialog.cboFemaleRatioAttribute.currentText() message = 'Expected %s but I got %s.' % (expected_female_attribute_key, female_attribute_key) self.assertEqual(expected_female_attribute_key, female_attribute_key, message) is_enabled = dialog.dsbFemaleRatioDefault.isEnabled() message = 'Expected disabled but I got enabled.' self.assertEqual(is_enabled, False, message)
def test_sum_ratio_behavior(self): """Test for wizard's behavior related sum of age ratio.""" layer = clone_shp_layer(name='kabupaten_jakarta', include_keywords=True, source_directory=BOUNDDATA) dialog = WizardDialog(layer=layer) dialog.suppress_warning_dialog = True self.check_current_text('aggregation', dialog.lstCategories) dialog.pbnNext.click() # Go to unit step self.check_current_text('KAB_NAME', dialog.lstFields) dialog.pbnNext.click() # Go to aggregation step dialog.dsbYouthRatioDefault.setValue(1.0) dialog.pbnNext.click() # Try to go to source step # check if still in aggregation step self.check_current_step(step_aggregation, dialog) dialog.cboYouthRatioAttribute.setCurrentIndex(1) # set don't use dialog.pbnNext.click() # Try to go to source step # check if in source step self.check_current_step(step_source, dialog) dialog.pbnBack.click() # Go to aggregation step # check if in aggregation step self.check_current_step(step_aggregation, dialog) dialog.cboYouthRatioAttribute.setCurrentIndex(0) # set global default dialog.dsbYouthRatioDefault.setValue(0.0) dialog.pbnNext.click() # Try to go to source step # check if in source step self.check_current_step(step_source, dialog) dialog.pbnCancel.click()
def test_point_layer(self): """Wizard for point layer.""" layer = clone_shp_layer( name='Marapi', include_keywords=True, source_directory=HAZDATA) dialog = WizardDialog(layer=layer) dialog.pbnNext.click() # choose hazard go to subcategory step dialog.pbnNext.click() # choose volcano go to source step # check if in step source self.check_current_step(step_source, dialog) dialog.pbnNext.click() # choose volcano go to title step # check if in step title self.check_current_step(step_title, dialog) dialog.accept()
def test_default_attributes_value(self): """Checking that default attributes is set to the CIA's one.""" layer = clone_shp_layer(name='kecamatan_jakarta', include_keywords=True, source_directory=BOUNDDATA) dialog = WizardDialog(layer=layer) dialog.pbnNext.click() # choose aggregation go to field step dialog.pbnNext.click() # choose KEC_NAME go to aggregation step ratio_attribute = dialog.cboFemaleRatioAttribute.currentText() message = 'Expected Global default but I got %s' % ratio_attribute self.assertEqual('Global default', ratio_attribute, message) ratio_attribute = dialog.cboElderlyRatioAttribute.currentText() message = 'Expected Global default but I got %s' % ratio_attribute self.assertEqual('Global default', ratio_attribute, message) ratio_attribute = dialog.cboAdultRatioAttribute.currentText() message = 'Expected Global default but I got %s' % ratio_attribute self.assertEqual('Global default', ratio_attribute, message) ratio_attribute = dialog.cboYouthRatioAttribute.currentText() message = 'Expected Global default but I got %s' % ratio_attribute self.assertEqual('Global default', ratio_attribute, message) default_value = dialog.dsbFemaleRatioDefault.value() expected_default_value = 0.50 message = ('Expected %s but I got %s' % (expected_default_value, default_value)) self.assertEqual(expected_default_value, default_value, message) default_value = dialog.dsbYouthRatioDefault.value() expected_default_value = 0.26 message = ('Expected %s but I got %s' % (expected_default_value, default_value)) self.assertEqual(expected_default_value, default_value, message) default_value = dialog.dsbAdultRatioDefault.value() expected_default_value = 0.66 message = ('Expected %s but I got %s' % (expected_default_value, default_value)) self.assertEqual(expected_default_value, default_value, message) default_value = dialog.dsbElderlyRatioDefault.value() expected_default_value = 0.08 message = ('Expected %s but I got %s' % (expected_default_value, default_value)) self.assertEqual(expected_default_value, default_value, message)
def test_integrated_line(self): """Test for line layer and all possibilities.""" layer = clone_shp_layer(name='jakarta_roads', source_directory=EXPDATA) dialog = WizardDialog(layer=layer) expected_categories = ['exposure'] self.check_list(expected_categories, dialog.lstCategories) self.check_current_text('exposure', dialog.lstCategories) dialog.pbnNext.click() # go to subcategory # check if in step subcategory self.check_current_step(step_subcategory, dialog) expected_subcategories = ['road'] self.check_list(expected_subcategories, dialog.lstSubcategories) self.check_current_text('road', dialog.lstSubcategories) dialog.pbnNext.click() # go to unit # check if in step unit self.check_current_step(step_unit, dialog) expected_units = ['Road Type'] self.check_list(expected_units, dialog.lstUnits) self.check_current_text(expected_units[0], dialog.lstUnits) dialog.pbnNext.click() # go to field # check if in step field self.check_current_step(step_field, dialog) expected_fields = ['TYPE', 'NAME', 'ONEWAY', 'LANES'] self.check_list(expected_fields, dialog.lstFields) # select Type self.select_from_list_widget('TYPE', dialog.lstFields) dialog.pbnNext.click() # go to source step dialog.pbnNext.click() # go to title step dialog.pbnCancel.click() # cancel
def test_unit_building_generic(self): """Test for case existing building generic unit for structure.""" layer = clone_shp_layer(name='building_Maumere', include_keywords=True, source_directory=TESTDATA) dialog = WizardDialog(layer=layer) dialog.pbnNext.click() # go to subcategory step 2 dialog.pbnNext.click() # go to unit step 3 dialog.lstUnits.setCurrentRow(1) # select no type dialog.pbnNext.click() # should be in step source # check if in step source self.check_current_step(step_source, dialog) dialog.pbnNext.click() # should be in step title # check if in step title self.check_current_step(step_title, dialog) dialog.pbnNext.click() # finishing
def test_auto_select_one_item(self): """Test auto select if there is only one item in a list.""" layer = clone_shp_layer(name='Marapi_evac_zone_3000m', include_keywords=True, source_directory=HAZDATA) dialog = WizardDialog(layer=layer) dialog.pbnNext.click() # choose hazard go to subcategory step dialog.pbnNext.click() # choose volcano go to unit step message = 'It should auto select, but it does not.' self.assertTrue(dialog.lstUnits.currentRow() == 0, message) num_item = dialog.lstUnits.count() message = 'There is should be only one item, I got %s' % num_item self.assertTrue(num_item == 1, message) dialog.pbnNext.click() # choose volcano go to field step message = 'It should auto select, but it does not.' self.assertTrue(dialog.lstFields.currentRow() == 0, message) num_item = dialog.lstFields.count() message = 'There is should be only one item, I got %s' % num_item self.assertTrue(num_item == 1, message)
def test_integrated_raster(self): """Test for raster layer and all possibilities.""" layer = clone_raster_layer(name='eq_yogya_2006', extension='.asc', include_keywords=False, source_directory=HAZDATA) dialog = WizardDialog(layer=layer) expected_categories = ['hazard', 'exposure'] self.check_list(expected_categories, dialog.lstCategories) # check if no option is selected expected_category = -1 categories = dialog.lstCategories.currentRow() message = ('Expected %s, but I got %s' % (expected_category, categories)) self.assertEqual(expected_category, categories, message) # choosing hazard self.select_from_list_widget('hazard', dialog.lstCategories) dialog.pbnNext.click() # Go to subcategory # check if in step subcategory self.check_current_step(step_subcategory, dialog) # check the values of subcategories options expected_subcategories = [ 'flood', 'tephra', 'volcano', 'earthquake', 'tsunami', 'generic' ] self.check_list(expected_subcategories, dialog.lstSubcategories) # check if no option is selected expected_subcategory_index = -1 subcategory_index = dialog.lstSubcategories.currentRow() message = ('Expected %s, but I got %s' % (expected_subcategory_index, subcategory_index)) self.assertEqual(expected_subcategory_index, subcategory_index, message) # choosing flood self.select_from_list_widget('flood', dialog.lstSubcategories) dialog.pbnNext.click() # Go to unit # check if in step unit self.check_current_step(step_unit, dialog) # check the values of units options expected_units = ['categorised', 'normalised', 'metres', 'feet'] self.check_list(expected_units, dialog.lstUnits) # choosing metres self.select_from_list_widget('metres', dialog.lstUnits) dialog.pbnNext.click() # Go to source # check if in step source self.check_current_step(step_source, dialog) dialog.pbnBack.click() # back to step unit dialog.pbnBack.click() # back to step subcategory # check if in step subcategory self.check_current_step(step_subcategory, dialog) # check if flood is selected expected_subcategory = 'flood' subcategory = dialog.lstSubcategories.currentItem().text() message = ('Expected %s, but I got %s' % (expected_subcategory, subcategory)) self.assertEqual(expected_subcategory, subcategory, message) # choosing earthquake self.select_from_list_widget('earthquake', dialog.lstSubcategories) dialog.pbnNext.click() # Go to unit # check if in step unit self.check_current_step(step_unit, dialog) # check the values of units options expected_units = ['categorised', 'normalised', 'MMI'] self.check_list(expected_units, dialog.lstUnits) # choosing MMI self.select_from_list_widget('MMI', dialog.lstUnits) dialog.pbnNext.click() # Go to source # check if in step source self.check_current_step(step_source, dialog)
def test_existing_complex_keywords(self): layer = clone_shp_layer(name='tsunami_polygon', include_keywords=True) # noinspection PyTypeChecker dialog = WizardDialog(layer=layer) # select hazard self.select_from_list_widget('hazard', dialog.lstCategories) dialog.pbnNext.click() # select volcano self.select_from_list_widget('volcano', dialog.lstSubcategories) dialog.pbnNext.click() # select volcano categorical unit self.select_from_list_widget('volcano categorical', dialog.lstUnits) dialog.pbnNext.click() # select GRIDCODE self.select_from_list_widget('GRIDCODE', dialog.lstFields) dialog.pbnNext.click() unit = dialog.selected_unit() default_classes = unit['classes'] unassigned_values = [] # no need to check actually, not save in file assigned_values = { 'low': ['5.0'], 'medium': ['3.0', '4.0'], 'high': ['2.0'] } dialog.populate_classified_values( unassigned_values, assigned_values, default_classes) dialog.pbnNext.click() source = 'Source' source_scale = 'Source Scale' source_url = 'Source Url' source_date = 'Source Date' dialog.leSource.setText(source) dialog.leSource_scale.setText(source_scale) dialog.leSource_url.setText(source_url) dialog.leSource_date.setText(source_date) dialog.pbnNext.click() # next dialog.pbnNext.click() # finish # noinspection PyTypeChecker dialog = WizardDialog(layer=layer) # step 1 of 7 - select category self.check_current_text('hazard', dialog.lstCategories) # Click Next dialog.pbnNext.click() # step 2 of 7 - select subcategory # noinspection PyTypeChecker self.check_current_text('volcano', dialog.lstSubcategories) # Click Next dialog.pbnNext.click() # step 3 of 7 - select volcano units self.check_current_text('volcano categorical', dialog.lstUnits) # Click Next dialog.pbnNext.click() # step 4 of 7 - select field self.check_current_text('GRIDCODE', dialog.lstFields) # Click Next dialog.pbnNext.click() for index in range(dialog.lstUniqueValues.count()): message = ('%s Should be in unassigned values' % dialog.lstUniqueValues.item(index).text()) self.assertIn( dialog.lstUniqueValues.item(index).text(), unassigned_values, message) real_assigned_values = dialog.selected_mapping() self.assertDictEqual(real_assigned_values, assigned_values) # Click Next dialog.pbnNext.click() # step 6 of 7 - enter source message = ('Invalid Next button state in step 6! Disabled while ' 'source is optional') self.assertTrue(dialog.pbnNext.isEnabled(), message) message = 'Source should be %s' % source self.assertEqual(dialog.leSource.text(), source, message) message = 'Source Url should be %s' % source_url self.assertEqual(dialog.leSource_url.text(), source_url, message) message = 'Source Date should be %s' % source_date self.assertEqual(dialog.leSource_date.text(), source_date, message) message = 'Source Scale should be %s' % source_scale self.assertEqual(dialog.leSource_scale.text(), source_scale, message) dialog.pbnNext.click() dialog.pbnCancel.click()
def test_keywords_creation_wizard(self): """Test how the widgets work.""" expected_category_count = 3 expected_categories = ['exposure', 'hazard', 'aggregation'] chosen_category = 'hazard' expected_subcategory_count = 4 expected_subcategories = ['volcano', 'earthquake', 'flood', 'tsunami'] chosen_subcategory = "tsunami" expected_unit_count = 3 expected_units = ['wetdry', 'metres_depth', 'feet_depth'] expected_chosen_unit = 'feet_depth' expected_field_count = 5 expected_fields = [ 'OBJECTID', 'GRIDCODE', 'Shape_Leng', 'Shape_Area', 'Category' ] expected_chosen_field = 'GRIDCODE' expected_keywords = { 'category': 'hazard', 'subcategory': 'tsunami', 'unit': 'feet_depth', 'field': 'GRIDCODE', 'source': 'some source', 'title': 'some title' } layer = clone_shp_layer(name='tsunami_polygon') # check the environment first message = 'Test layer is not readable. Check environment variables.' self.assertIsNotNone(layer.dataProvider(), message) # Initialize dialog # noinspection PyTypeChecker dialog = WizardDialog(layer=layer) # step 1 of 7 - select category count = dialog.lstCategories.count() message = ('Invalid category count! There should be %d while there ' 'were: %d') % (expected_category_count, count) self.assertEqual(count, expected_category_count, message) # Get all the categories given by wizards and save the 'hazard' index categories = [] hazard_index = -1 for i in range(expected_category_count): category_name = eval( dialog.lstCategories.item(i).data(Qt.UserRole))['id'] categories.append(category_name) if category_name == chosen_category: hazard_index = i # Check if categories is the same with expected_categories message = 'Invalid categories! It should be "%s" while it was %s' % ( expected_categories, categories) self.assertEqual(set(categories), set(expected_categories), message) # Check if the Next button state is on the right state message = ('Invalid Next button state in step 1! Enabled while ' 'there\'s nothing selected yet') self.assertTrue(not dialog.pbnNext.isEnabled(), message) # Select hazard one dialog.lstCategories.setCurrentRow(hazard_index) message = ('Invalid Next button state in step 1! Still disabled after ' 'an item selected') self.assertTrue(dialog.pbnNext.isEnabled(), message) # Click Next dialog.pbnNext.click() # step 2 of 7 - select subcategory # Check the number of sub categories count = dialog.lstSubcategories.count() message = ('Invalid subcategory count! There should be %d and there ' 'were: %d') % (expected_subcategory_count, count) self.assertEqual(count, expected_subcategory_count, message) # Get all the subcategories given and save the 'tsunami' index subcategories = [] tsunami_index = -1 for i in range(expected_subcategory_count): subcategory_name = eval( dialog.lstSubcategories.item(i).data(Qt.UserRole))['id'] subcategories.append(subcategory_name) if subcategory_name == chosen_subcategory: tsunami_index = i # Check if subcategories is the same with expected_subcategories message = ('Invalid sub categories! It should be "%s" while it was ' '%s') % (expected_subcategories, subcategories) self.assertEqual(set(subcategories), set(expected_subcategories), message) # The Next button should be on disabled state first self.assertTrue( not dialog.pbnNext.isEnabled(), 'Invalid Next button' ' state in step 2! Enabled while there\'s nothing selected yet') # Set to tsunami subcategories dialog.lstSubcategories.setCurrentRow(tsunami_index) message = ('Invalid Next button state in step 2! Still disabled after ' 'an item selected') self.assertTrue(dialog.pbnNext.isEnabled(), message) # Click next button dialog.pbnNext.click() # step 3 of 7 - select tsunami units # Check if the number of unit for tsunami is 3 count = dialog.lstUnits.count() message = ('Invalid unit count! There should be %d while there were: ' '%d') % (expected_unit_count, count) self.assertEqual(count, expected_unit_count, message) # Get all the units given and save the 'feet_depth' index units = [] feet_unit_index = -1 for i in range(expected_unit_count): unit_name = eval(dialog.lstUnits.item(i).data(Qt.UserRole))['id'] units.append(unit_name) if unit_name == expected_chosen_unit: feet_unit_index = i # Check if units is the same with expected_units message = ('Invalid units! It should be "%s" while it was ' '%s') % (expected_units, units) self.assertEqual(set(expected_units), set(units), message) # The button should be on disabled state first message = ('Invalid Next button state in step 3! Enabled while ' 'there\'s nothing selected yet') self.assertTrue(not dialog.pbnNext.isEnabled(), message) dialog.lstUnits.setCurrentRow(feet_unit_index) message = ('Invalid Next button state in step 3! Enabled while ' 'there\'s nothing selected yet') self.assertTrue(dialog.pbnNext.isEnabled(), message) dialog.pbnNext.click() # step 4 of 7 - select data field for tsunami feet count = dialog.lstFields.count() message = ('Invalid field count! There should be %d while there were: ' '%d') % (expected_field_count, count) self.assertEqual(count, expected_field_count, message) # Get all the fields given and save the 'GRIDCODE' index fields = [] gridcode_index = -1 for i in range(expected_field_count): field_name = dialog.lstFields.item(i).text() fields.append(field_name) if field_name == expected_chosen_field: gridcode_index = i # Check if fields is the same with expected_fields message = ('Invalid fields! It should be "%s" while it was ' '%s') % (expected_fields, fields) self.assertEqual(set(expected_fields), set(fields), message) # The button should be on disabled first message = ('Invalid Next button state in step 4! Enabled while ' 'there\'s nothing selected yet') self.assertTrue(not dialog.pbnNext.isEnabled(), message) dialog.lstFields.setCurrentRow(gridcode_index) message = ('Invalid Next button state in step 4! Still disabled after ' 'an item selected') self.assertTrue(dialog.pbnNext.isEnabled(), message) # Click next dialog.pbnNext.click() # step 6 of 7 - enter source message = ('Invalid Next button state in step 6! Disabled while ' 'source is optional') self.assertTrue(dialog.pbnNext.isEnabled(), message) dialog.leSource.setText('some source') dialog.pbnNext.click() # step 7 of 7 - enter title dialog.leTitle.setText('some title') message = ('Invalid Next button state in step 7! Still disabled ' 'after a text entered') self.assertTrue(dialog.pbnNext.isEnabled(), message) dialog.pbnNext.click() # test the resulting keywords keyword_io = KeywordIO() # noinspection PyTypeChecker keywords = keyword_io.read_keywords(layer) message = 'Invalid metadata!\n Was: %s\n Should be: %s' % ( unicode(keywords), unicode(expected_keywords)) self.assertEqual(keywords, expected_keywords, message)
def test_existing_keywords(self): """Test if keywords are already exist.""" expected_field_count = 5 expected_fields = [ 'OBJECTID', 'GRIDCODE', 'Shape_Leng', 'Shape_Area', 'Category' ] expected_chosen_field = 'GRIDCODE' layer = clone_shp_layer(name='tsunami_polygon', include_keywords=True) # check the environment first message = 'Test layer is not readable. Check environment variables.' self.assertIsNotNone(layer.dataProvider(), message) # Initialize dialog # noinspection PyTypeChecker dialog = WizardDialog(layer=layer) # step 1 of 7 - select category self.check_current_text('hazard', dialog.lstCategories) message = ('Invalid Next button state in step 1! Still disabled after ' 'an item selected') self.assertTrue(dialog.pbnNext.isEnabled(), message) # Click Next dialog.pbnNext.click() # step 2 of 7 - select subcategory # noinspection PyTypeChecker self.check_current_text('tsunami', dialog.lstSubcategories) message = ('Invalid Next button state in step 2! Still disabled after ' 'an item selected') self.assertTrue(dialog.pbnNext.isEnabled(), message) # Click Next dialog.pbnNext.click() # step 3 of 7 - select tsunami units self.check_current_text('metres', dialog.lstUnits) message = ('Invalid Next button state in step 2! Still disabled after ' 'an item selected') self.assertTrue(dialog.pbnNext.isEnabled(), message) # Click Next dialog.pbnNext.click() # step 4 of 7 - select data field for tsunami feet count = dialog.lstFields.count() message = ('Invalid field count! There should be %d while there were: ' '%d') % (expected_field_count, count) self.assertEqual(count, expected_field_count, message) # Get all the fields given and save the 'GRIDCODE' index fields = [] gridcode_index = -1 for i in range(expected_field_count): field_name = dialog.lstFields.item(i).text() fields.append(field_name) if field_name == expected_chosen_field: gridcode_index = i # Check if fields is the same with expected_fields message = ('Invalid fields! It should be "%s" while it was ' '%s') % (expected_fields, fields) self.assertEqual(set(expected_fields), set(fields), message) # The button should be on disabled first message = ('Invalid Next button state in step 4! Enabled while ' 'there\'s nothing selected yet') self.assertTrue(not dialog.pbnNext.isEnabled(), message) dialog.lstFields.setCurrentRow(gridcode_index) message = ('Invalid Next button state in step 4! Still disabled after ' 'an item selected') self.assertTrue(dialog.pbnNext.isEnabled(), message) # Click next dialog.pbnNext.click() # step 6 of 7 - enter source message = ('Invalid Next button state in step 6! Disabled while ' 'source is optional') self.assertTrue(dialog.pbnNext.isEnabled(), message) message = 'Source should be empty' self.assertEqual(dialog.leSource.text(), '', message) message = 'Source Url should be empty' self.assertEqual(dialog.leSource_url.text(), '', message) message = 'Source Date should be empty' self.assertEqual(dialog.leSource_date.text(), '', message) message = 'Source Scale should be empty' self.assertEqual(dialog.leSource_scale.text(), '', message) dialog.pbnNext.click() # step 7 of 7 - enter title message = 'Title should be %s but I got %s' % (dialog.layer.name(), dialog.leTitle.text()) self.assertEqual(dialog.layer.name(), dialog.leTitle.text(), message) message = ('Invalid Next button state in step 7! Still disabled ' 'after a text entered') self.assertTrue(dialog.pbnNext.isEnabled(), message) dialog.pbnNext.click()
def test_existing_complex_keywords(self): layer = clone_shp_layer(include_keywords=True) # noinspection PyTypeChecker dialog = WizardDialog(PARENT, IFACE, None, layer) # All desired item chosen_category = 'hazard' chosen_sub_category = 'volcano' chosen_unit = 'volcano categorical' chosen_field = 'GRIDCODE' # Get index for hazard category chosen_category_index = -1 for i in range(dialog.lstCategories.count()): category_name = dialog.lstCategories.item(i).text() if category_name == chosen_category: chosen_category_index = i message = 'There is no %s in the list widget of category.' % ( chosen_category) self.assertIsNot(chosen_category_index, -1, message) # Choose the row and click! dialog.lstCategories.setCurrentRow(chosen_category_index) dialog.pbnNext.click() # Get index for sub category volcano chosen_sub_category_index = -1 for i in range(dialog.lstSubcategories.count()): sub_category_name = dialog.lstSubcategories.item(i).text() if sub_category_name == chosen_sub_category: chosen_sub_category_index = i message = 'There is no %s in the list widget of sub category.' % ( chosen_sub_category) self.assertIsNot(chosen_sub_category_index, -1, message) # Choose the row and click! dialog.lstSubcategories.setCurrentRow(chosen_sub_category_index) dialog.pbnNext.click() # Get index for volcano categorical unit chosen_unit_index = -1 for i in range(dialog.lstUnits.count()): unit_name = dialog.lstUnits.item(i).text() if unit_name == chosen_unit: chosen_unit_index = i message = 'There is no %s in the list widget of unit.' % ( chosen_unit) self.assertIsNot(chosen_unit_index, -1, message) # Choose the row and click! dialog.lstUnits.setCurrentRow(chosen_unit_index) dialog.pbnNext.click() # Get index for GRIDCODE field chosen_field_index = -1 for i in range(dialog.lstFields.count()): field_name = dialog.lstFields.item(i).text() if field_name == chosen_field: chosen_field_index = i message = 'There is no %s in the list widget of field.' % ( chosen_field) self.assertIsNot(chosen_field_index, -1, message) # Choose the row and click! dialog.lstFields.setCurrentRow(chosen_field_index) dialog.pbnNext.click() unit = dialog.selected_unit() default_classes = unit['classes'] unassigned_values = [] # no need to check actually, not save in # file assigned_values = { 'high': ['4.0', '5.0'], 'medium': ['3.0'], 'low': ['2.0'] } dialog.populate_classified_values( unassigned_values, assigned_values, default_classes) dialog.pbnNext.click() source = 'Source' source_scale = 'Source Scale' source_url = 'Source Url' source_date = 'Source Date' dialog.leSource.setText(source) dialog.leSource_scale.setText(source_scale) dialog.leSource_url.setText(source_url) dialog.leSource_date.setText(source_date) dialog.pbnNext.click() # next dialog.pbnNext.click() # finish # noinspection PyTypeChecker dialog = WizardDialog(PARENT, IFACE, None, layer) # step 1 of 7 - select category expected_category_text = 'hazard' category_text = dialog.lstCategories.currentItem().text() message = 'Expect category text %s but I got %s' % ( expected_category_text, category_text) self.assertEqual(expected_category_text, category_text, message) # Click Next dialog.pbnNext.click() # step 2 of 7 - select subcategory # noinspection PyTypeChecker expected_subcategory_text = 'volcano' subcategory_text = dialog.lstSubcategories.currentItem().text() message = 'Expect subcategory text %s but I got %s' % ( expected_subcategory_text, subcategory_text) self.assertEqual(expected_subcategory_text, subcategory_text, message) # Click Next dialog.pbnNext.click() # step 3 of 7 - select volcano units expected_unit_text = 'volcano categorical' unit_text = dialog.lstUnits.currentItem().text() message = 'Expect unit text %s but I got %s' % ( expected_unit_text, unit_text) self.assertEqual(expected_unit_text, unit_text, message) # Click Next dialog.pbnNext.click() # step 4 of 7 - select field expected_unit_text = 'GRIDCODE' unit_text = dialog.lstFields.currentItem().text() message = 'Expect unit text %s but I got %s' % ( expected_unit_text, unit_text) self.assertEqual(expected_unit_text, unit_text, message) # Click Next dialog.pbnNext.click() for index in range(dialog.lstUniqueValues.count()): message = ('%s Should be in unassigned values' % dialog.lstUniqueValues.item(index).text()) self.assertIn( dialog.lstUniqueValues.item(index).text(), unassigned_values, message) real_assigned_values = dialog.selected_mapping() self.assertDictEqual(real_assigned_values, assigned_values) # Click Next dialog.pbnNext.click() # step 6 of 7 - enter source message = ('Invalid Next button state in step 6! Disabled while ' 'source is optional') self.assertTrue(dialog.pbnNext.isEnabled(), message) message = 'Source should be %s' % source self.assertEqual(dialog.leSource.text(), source, message) message = 'Source Url should be %s' % source_url self.assertEqual(dialog.leSource_url.text(), source_url, message) message = 'Source Date should be %s' % source_date self.assertEqual(dialog.leSource_date.text(), source_date, message) message = 'Source Scale should be %s' % source_scale self.assertEqual(dialog.leSource_scale.text(), source_scale, message) dialog.pbnNext.click() dialog.pbnCancel.click() remove_temp_file(layer.source())
def test_integrated_polygon(self): """Test for polygon layer and all possibilities.""" layer = clone_shp_layer(name='Jakarta_RW_2007flood', source_directory=HAZDATA, include_keywords=False) dialog = WizardDialog(layer=layer) expected_categories = ['hazard', 'exposure', 'aggregation'] self.check_list(expected_categories, dialog.lstCategories) # choosing exposure self.select_from_list_widget('exposure', dialog.lstCategories) dialog.pbnNext.click() # Go to subcategory # check number of subcategories expected_subcategories = ['structure'] self.check_list(expected_subcategories, dialog.lstSubcategories) # check if automatically select the only option self.check_current_text(expected_subcategories[0], dialog.lstSubcategories) dialog.pbnNext.click() # Go to unit # check if in step unit self.check_current_step(step_unit, dialog) # check the values of units options expected_units = ['building type', 'building generic'] self.check_list(expected_units, dialog.lstUnits) # choosing building type self.select_from_list_widget('building type', dialog.lstUnits) dialog.pbnNext.click() # Go to field # check if in step field self.check_current_step(step_field, dialog) # check the values of field options expected_fields = [ 'KAB_NAME', 'KEC_NAME', 'KEL_NAME', 'RW', 'FLOODPRONE' ] self.check_list(expected_fields, dialog.lstFields) # choosing KAB_NAME self.select_from_list_widget('KAB_NAME', dialog.lstFields) dialog.pbnNext.click() # Go to source # check if in source step self.check_current_step(step_source, dialog) dialog.pbnBack.click() # back to field step dialog.pbnBack.click() # back to unit step # choosing building generic self.select_from_list_widget('building generic', dialog.lstUnits) dialog.pbnNext.click() # Go to source # check if in source source self.check_current_step(step_source, dialog) dialog.pbnBack.click() # back to unit step dialog.pbnBack.click() # back to subcategory step dialog.pbnBack.click() # back to category step # choosing hazard self.select_from_list_widget('hazard', dialog.lstCategories) dialog.pbnNext.click() # Go to subcategory # check the values of subcategories options expected_subcategories = ['earthquake', 'flood', 'tsunami', 'volcano'] self.check_list(expected_subcategories, dialog.lstSubcategories) # choosing earthquake self.select_from_list_widget('earthquake', dialog.lstSubcategories) dialog.pbnNext.click() # Go to unit # check the values of units options expected_units = ['MMI'] self.check_list(expected_units, dialog.lstUnits) self.select_from_list_widget('MMI', dialog.lstUnits) dialog.pbnNext.click() # go to field step # check in field step self.check_current_step(step_field, dialog) for i in range(dialog.lstFields.count()): item_flag = dialog.lstFields.item(i).flags() message = 'Item should be disabled' self.assertTrue(item_flag & ~QtCore.Qt.ItemIsEnabled, message) dialog.pbnBack.click() # back to unit step dialog.pbnBack.click() # back to unit subcategory # select flood self.select_from_list_widget('flood', dialog.lstSubcategories) dialog.pbnNext.click() # go to unit self.check_current_step(step_unit, dialog) expected_units = ['wet / dry', 'metres', 'feet'] self.check_list(expected_units, dialog.lstUnits) # select wet / dry self.select_from_list_widget('wet / dry', dialog.lstUnits) dialog.pbnNext.click() # go to fields self.check_current_step(step_field, dialog) expected_fields = [ 'KAB_NAME', 'KEC_NAME', 'KEL_NAME', 'RW', 'FLOODPRONE' ] self.check_list(expected_fields, dialog.lstFields) # select FLOODPRONE self.select_from_list_widget('FLOODPRONE', dialog.lstFields) dialog.pbnNext.click() # go to classify self.check_current_step(step_classify, dialog) # check unclassified expected_unique_values = ['Yes'] # Unclassified value self.check_list(expected_unique_values, dialog.lstUniqueValues) # check classified root = dialog.treeClasses.invisibleRootItem() expected_classes = ['wet', 'dry'] child_count = root.childCount() message = 'Child count must be %s' % len(expected_classes) self.assertEqual(len(expected_classes), child_count, message) for i in range(child_count): item = root.child(i) class_name = item.text(0) message = '%s should not be in classes name' % class_name self.assertIn(class_name, expected_classes, message) if class_name == 'wet': expected_num_child = 1 num_child = item.childCount() message = 'The child of wet should be %s' % expected_num_child self.assertEqual(expected_num_child, num_child, message) if class_name == 'dry': expected_num_child = 0 num_child = item.childCount() message = 'The child of dry should be %s' % expected_num_child self.assertEqual(expected_num_child, num_child, message) dialog.pbnNext.click() # go to source self.check_current_step(step_source, dialog) dialog.pbnBack.click() # back to classify dialog.pbnBack.click() # back to field dialog.pbnBack.click() # back to unit self.select_from_list_widget('metres', dialog.lstUnits) dialog.pbnNext.click() # go to field # check in field step self.check_current_step(step_field, dialog) # check if all options are disabled for i in range(dialog.lstFields.count()): item_flag = dialog.lstFields.item(i).flags() message = 'Item should be disabled' self.assertTrue(item_flag & ~QtCore.Qt.ItemIsEnabled, message) dialog.pbnBack.click() # back to unit dialog.pbnBack.click() # back to subcategory self.select_from_list_widget('tsunami', dialog.lstSubcategories) dialog.pbnNext.click() # go to unit self.check_current_step(step_unit, dialog) # back again since tsunami similar to flood dialog.pbnBack.click() # back to subcategory self.select_from_list_widget('volcano', dialog.lstSubcategories) dialog.pbnNext.click() # go to unit expected_units = ['volcano categorical'] self.check_list(expected_units, dialog.lstUnits) # no need to select, use auto select dialog.pbnNext.click() # go to field self.check_current_step(step_field, dialog) self.select_from_list_widget('FLOODPRONE', dialog.lstFields) dialog.pbnNext.click() # go to classify self.check_current_step(step_classify, dialog) # check unclassified expected_unique_values = ['Yes', 'YES'] # Unclassified value self.check_list(expected_unique_values, dialog.lstUniqueValues) # check classified root = dialog.treeClasses.invisibleRootItem() expected_classes = ['low', 'medium', 'high'] child_count = root.childCount() message = 'Child count must be %s' % len(expected_classes) self.assertEqual(len(expected_classes), child_count, message) for i in range(child_count): item = root.child(i) class_name = item.text(0) message = '%s should not be in classes name' % class_name self.assertIn(class_name, expected_classes, message) expected_num_child = 0 num_child = item.childCount() message = 'The child of wet should be %s' % expected_num_child self.assertEqual(expected_num_child, num_child, message) dialog.pbnNext.click() # go to source self.check_current_step(step_source, dialog) dialog.pbnCancel.click()
def test_existing_complex_keywords(self): """Test for existing complex keywords in wizard in locale mode.""" from safe_qgis.tools.wizard_dialog import WizardDialog from safe_qgis.utilities.utilities_for_testing import ( clone_shp_layer, remove_vector_temp_file) layer = clone_shp_layer(include_keywords=True) from safe.common.testing import get_qgis_app # Get QGis app handle # noinspection PyPep8Naming _, _, IFACE, PARENT = get_qgis_app() # noinspection PyTypeChecker dialog = WizardDialog(PARENT, IFACE, None, layer) # select hazard self.select_from_list_widget('ancaman', dialog.lstCategories) dialog.pbnNext.click() # select volcano self.select_from_list_widget('gunung berapi', dialog.lstSubcategories) dialog.pbnNext.click() # select volcano categorical unit self.select_from_list_widget('Kategori gunung berapi', dialog.lstUnits) dialog.pbnNext.click() # select GRIDCODE self.select_from_list_widget('GRIDCODE', dialog.lstFields) dialog.pbnNext.click() unit = dialog.selected_unit() default_classes = unit['classes'] unassigned_values = [] # no need to check actually, not save in file assigned_values = { 'low': ['5.0'], 'medium': ['3.0', '4.0'], 'high': ['2.0'] } dialog.populate_classified_values( unassigned_values, assigned_values, default_classes) dialog.pbnNext.click() source = 'Source' source_scale = 'Source Scale' source_url = 'Source Url' source_date = 'Source Date' dialog.leSource.setText(source) dialog.leSource_scale.setText(source_scale) dialog.leSource_url.setText(source_url) dialog.leSource_date.setText(source_date) dialog.pbnNext.click() # next dialog.pbnNext.click() # finish # noinspection PyTypeChecker dialog = WizardDialog(PARENT, IFACE, None, layer) # step 1 of 7 - select category self.check_current_text('ancaman', dialog.lstCategories) # Click Next dialog.pbnNext.click() # step 2 of 7 - select subcategory # noinspection PyTypeChecker self.check_current_text('gunung berapi', dialog.lstSubcategories) # Click Next dialog.pbnNext.click() # step 3 of 7 - select volcano units self.check_current_text('Kategori gunung berapi', dialog.lstUnits) # Click Next dialog.pbnNext.click() # step 4 of 7 - select field self.check_current_text('GRIDCODE', dialog.lstFields) # Click Next dialog.pbnNext.click() for index in range(dialog.lstUniqueValues.count()): message = ('%s Should be in unassigned values' % dialog.lstUniqueValues.item(index).text()) self.assertIn( dialog.lstUniqueValues.item(index).text(), unassigned_values, message) real_assigned_values = dialog.selected_mapping() self.assertDictEqual(real_assigned_values, assigned_values) # Click Next dialog.pbnNext.click() # step 6 of 7 - enter source message = ('Invalid Next button state in step 6! Disabled while ' 'source is optional') self.assertTrue(dialog.pbnNext.isEnabled(), message) message = 'Source should be %s' % source self.assertEqual(dialog.leSource.text(), source, message) message = 'Source Url should be %s' % source_url self.assertEqual(dialog.leSource_url.text(), source_url, message) message = 'Source Date should be %s' % source_date self.assertEqual(dialog.leSource_date.text(), source_date, message) message = 'Source Scale should be %s' % source_scale self.assertEqual(dialog.leSource_scale.text(), source_scale, message) dialog.pbnNext.click() dialog.pbnCancel.click() remove_vector_temp_file(layer.source())
def test_existing_complex_keywords(self): """Test for existing complex keywords in wizard in locale mode.""" from safe_qgis.tools.wizard_dialog import WizardDialog from safe_qgis.utilities.utilities_for_testing import ( clone_shp_layer, remove_temp_file) layer = clone_shp_layer(include_keywords=True) from safe.common.testing import get_qgis_app # Get QGis app handle # noinspection PyPep8Naming _, _, IFACE, PARENT = get_qgis_app() # noinspection PyTypeChecker dialog = WizardDialog(PARENT, IFACE, None, layer) # select hazard self.select_from_list_widget('ancaman', dialog.lstCategories) dialog.pbnNext.click() # select volcano self.select_from_list_widget('gunung berapi', dialog.lstSubcategories) dialog.pbnNext.click() # select volcano categorical unit self.select_from_list_widget('Kategori gunung berapi', dialog.lstUnits) dialog.pbnNext.click() # select GRIDCODE self.select_from_list_widget('GRIDCODE', dialog.lstFields) dialog.pbnNext.click() unit = dialog.selected_unit() default_classes = unit['classes'] unassigned_values = [] # no need to check actually, not save in file assigned_values = { 'low': ['5.0'], 'medium': ['3.0', '4.0'], 'high': ['2.0'] } dialog.populate_classified_values( unassigned_values, assigned_values, default_classes) dialog.pbnNext.click() source = 'Source' source_scale = 'Source Scale' source_url = 'Source Url' source_date = 'Source Date' dialog.leSource.setText(source) dialog.leSource_scale.setText(source_scale) dialog.leSource_url.setText(source_url) dialog.leSource_date.setText(source_date) dialog.pbnNext.click() # next dialog.pbnNext.click() # finish # noinspection PyTypeChecker dialog = WizardDialog(PARENT, IFACE, None, layer) # step 1 of 7 - select category self.check_current_text('ancaman', dialog.lstCategories) # Click Next dialog.pbnNext.click() # step 2 of 7 - select subcategory # noinspection PyTypeChecker self.check_current_text('gunung berapi', dialog.lstSubcategories) # Click Next dialog.pbnNext.click() # step 3 of 7 - select volcano units self.check_current_text('Kategori gunung berapi', dialog.lstUnits) # Click Next dialog.pbnNext.click() # step 4 of 7 - select field self.check_current_text('GRIDCODE', dialog.lstFields) # Click Next dialog.pbnNext.click() for index in range(dialog.lstUniqueValues.count()): message = ('%s Should be in unassigned values' % dialog.lstUniqueValues.item(index).text()) self.assertIn( dialog.lstUniqueValues.item(index).text(), unassigned_values, message) real_assigned_values = dialog.selected_mapping() self.assertDictEqual(real_assigned_values, assigned_values) # Click Next dialog.pbnNext.click() # step 6 of 7 - enter source message = ('Invalid Next button state in step 6! Disabled while ' 'source is optional') self.assertTrue(dialog.pbnNext.isEnabled(), message) message = 'Source should be %s' % source self.assertEqual(dialog.leSource.text(), source, message) message = 'Source Url should be %s' % source_url self.assertEqual(dialog.leSource_url.text(), source_url, message) message = 'Source Date should be %s' % source_date self.assertEqual(dialog.leSource_date.text(), source_date, message) message = 'Source Scale should be %s' % source_scale self.assertEqual(dialog.leSource_scale.text(), source_scale, message) dialog.pbnNext.click() dialog.pbnCancel.click() remove_temp_file(layer.source())
def test_existing_complex_keywords(self): layer = clone_shp_layer(name='tsunami_polygon', include_keywords=True) # noinspection PyTypeChecker dialog = WizardDialog(layer=layer) # select hazard self.select_from_list_widget('hazard', dialog.lstCategories) dialog.pbnNext.click() # select volcano self.select_from_list_widget('volcano', dialog.lstSubcategories) dialog.pbnNext.click() # select volcano categorical unit self.select_from_list_widget('volcano categorical', dialog.lstUnits) dialog.pbnNext.click() # select GRIDCODE self.select_from_list_widget('GRIDCODE', dialog.lstFields) dialog.pbnNext.click() unit = dialog.selected_unit() default_classes = unit['classes'] unassigned_values = [] # no need to check actually, not save in file assigned_values = { 'low': ['5.0'], 'medium': ['3.0', '4.0'], 'high': ['2.0'] } dialog.populate_classified_values(unassigned_values, assigned_values, default_classes) dialog.pbnNext.click() source = 'Source' source_scale = 'Source Scale' source_url = 'Source Url' source_date = 'Source Date' dialog.leSource.setText(source) dialog.leSource_scale.setText(source_scale) dialog.leSource_url.setText(source_url) dialog.leSource_date.setText(source_date) dialog.pbnNext.click() # next dialog.pbnNext.click() # finish # noinspection PyTypeChecker dialog = WizardDialog(layer=layer) # step 1 of 7 - select category self.check_current_text('hazard', dialog.lstCategories) # Click Next dialog.pbnNext.click() # step 2 of 7 - select subcategory # noinspection PyTypeChecker self.check_current_text('volcano', dialog.lstSubcategories) # Click Next dialog.pbnNext.click() # step 3 of 7 - select volcano units self.check_current_text('volcano categorical', dialog.lstUnits) # Click Next dialog.pbnNext.click() # step 4 of 7 - select field self.check_current_text('GRIDCODE', dialog.lstFields) # Click Next dialog.pbnNext.click() for index in range(dialog.lstUniqueValues.count()): message = ('%s Should be in unassigned values' % dialog.lstUniqueValues.item(index).text()) self.assertIn( dialog.lstUniqueValues.item(index).text(), unassigned_values, message) real_assigned_values = dialog.selected_mapping() self.assertDictEqual(real_assigned_values, assigned_values) # Click Next dialog.pbnNext.click() # step 6 of 7 - enter source message = ('Invalid Next button state in step 6! Disabled while ' 'source is optional') self.assertTrue(dialog.pbnNext.isEnabled(), message) message = 'Source should be %s' % source self.assertEqual(dialog.leSource.text(), source, message) message = 'Source Url should be %s' % source_url self.assertEqual(dialog.leSource_url.text(), source_url, message) message = 'Source Date should be %s' % source_date self.assertEqual(dialog.leSource_date.text(), source_date, message) message = 'Source Scale should be %s' % source_scale self.assertEqual(dialog.leSource_scale.text(), source_scale, message) dialog.pbnNext.click() dialog.pbnCancel.click()