def show_impact_merge(self):
"""Show the impact layer merge dialog."""
# import here only so that it is AFTER i18n set up
from safe_qgis.tools.impact_merge_dialog import ImpactMergeDialog
dialog = ImpactMergeDialog(self.iface.mainWindow())
dialog.exec_() # modal
def setUp(self):
"""Runs before each test."""
#noinspection PyArgumentList
self.map_layer_registry = QgsMapLayerRegistry.instance()
self.register_layers()
# Create Impact Merge Dialog
self.impact_merge_dialog = ImpactMergeDialog(PARENT, IFACE)
# Create test dir
#noinspection PyUnresolvedReferences
test_impact_merge_dir = os.path.join(TEST_DATA_DIR,
'test-impact-merge')
if not os.path.exists(test_impact_merge_dir):
os.makedirs(test_impact_merge_dir)
# Create test dir for aggregated
#noinspection PyUnresolvedReferences
test_aggregated_dir = os.path.join(test_impact_merge_dir, 'aggregated')
if not os.path.exists(test_aggregated_dir):
os.makedirs(test_aggregated_dir)
# Create test dir for entire
test_entire_dir = os.path.join(test_impact_merge_dir, 'entire')
if not os.path.exists(test_entire_dir):
os.makedirs(test_entire_dir)
def show_impact_merge(self):
"""Show the impact layer merge dialog."""
# import here only so that it is AFTER i18n set up
from safe_qgis.tools.impact_merge_dialog import ImpactMergeDialog
dialog = ImpactMergeDialog(self.iface.mainWindow())
dialog.exec_() # modal
def setUp(self):
"""Runs before each test."""
#noinspection PyArgumentList
self.map_layer_registry = QgsMapLayerRegistry.instance()
self.register_layers()
# Create Impact Merge Dialog
self.impact_merge_dialog = ImpactMergeDialog(PARENT, IFACE)
# Create test dir
#noinspection PyUnresolvedReferences
test_impact_merge_dir = os.path.join(
TEST_DATA_DIR, 'test-impact-merge')
if not os.path.exists(test_impact_merge_dir):
os.makedirs(test_impact_merge_dir)
# Create test dir for aggregated
#noinspection PyUnresolvedReferences
test_aggregated_dir = os.path.join(
test_impact_merge_dir, 'aggregated')
if not os.path.exists(test_aggregated_dir):
os.makedirs(test_aggregated_dir)
# Create test dir for entire
test_entire_dir = os.path.join(
test_impact_merge_dir, 'entire')
if not os.path.exists(test_entire_dir):
os.makedirs(test_entire_dir)
class ImpactMergeDialogTest(unittest.TestCase):
"""Test Impact Merge Dialog widget."""
#noinspection PyPep8Naming
def setUp(self):
"""Runs before each test."""
#noinspection PyArgumentList
self.map_layer_registry = QgsMapLayerRegistry.instance()
self.register_layers()
# Create Impact Merge Dialog
self.impact_merge_dialog = ImpactMergeDialog(PARENT, IFACE)
# Create test dir
#noinspection PyUnresolvedReferences
test_impact_merge_dir = os.path.join(TEST_DATA_DIR,
'test-impact-merge')
if not os.path.exists(test_impact_merge_dir):
os.makedirs(test_impact_merge_dir)
# Create test dir for aggregated
#noinspection PyUnresolvedReferences
test_aggregated_dir = os.path.join(test_impact_merge_dir, 'aggregated')
if not os.path.exists(test_aggregated_dir):
os.makedirs(test_aggregated_dir)
# Create test dir for entire
test_entire_dir = os.path.join(test_impact_merge_dir, 'entire')
if not os.path.exists(test_entire_dir):
os.makedirs(test_entire_dir)
#noinspection PyPep8Naming
def tearDown(self):
"""Runs after each test."""
# Remove Map Layers
if len(self.map_layer_registry.mapLayers().values()) > 0:
self.map_layer_registry.removeAllMapLayers()
# Delete test dir
#noinspection PyUnresolvedReferences
test_impact_merge_dir = os.path.join(TEST_DATA_DIR,
'test-impact-merge')
shutil.rmtree(test_impact_merge_dir)
def register_layers(self):
"""Register needed layers to QgsMapLayerRegistry."""
# Register 4 impact layers and aggregation layer
self.population_entire_jakarta_layer, _ = load_layer(
population_entire_jakarta_impact_path, directory=None)
self.building_entire_jakarta_layer, _ = load_layer(
building_entire_jakarta_impact_path, directory=None)
self.population_district_jakarta_layer, _ = load_layer(
population_district_jakarta_impact_path, directory=None)
self.building_district_jakarta_layer, _ = load_layer(
building_district_jakarta_impact_path, directory=None)
self.district_jakarta_layer, _ = load_layer(
district_jakarta_boundary_path, directory=None)
layer_list = [
self.population_entire_jakarta_layer,
self.population_district_jakarta_layer,
self.building_entire_jakarta_layer,
self.building_district_jakarta_layer, self.district_jakarta_layer
]
# noinspection PyArgumentList
self.map_layer_registry.addMapLayers(layer_list)
def mock_the_dialog(self, test_entire_mode):
impact_layer_count = self.impact_merge_dialog.first_layer.count()
aggregation_layer_count = \
self.impact_merge_dialog.aggregation_layer.count()
if test_entire_mode:
# First impact layer = population entire
# Second impact layer = buildings entire
self.impact_merge_dialog.entire_area_mode = True
# Set the current Index of the combobox
for index in range(0, impact_layer_count):
first_combobox = self.impact_merge_dialog.first_layer
layer_name = first_combobox.itemText(index)
if ('population' in layer_name) and ('entire' in layer_name):
self.impact_merge_dialog.first_layer.setCurrentIndex(index)
for index in range(0, impact_layer_count):
second_combobox = self.impact_merge_dialog.second_layer
layer_name = second_combobox.itemText(index)
if ('buildings' in layer_name) and ('entire' in layer_name):
self.impact_merge_dialog.second_layer.setCurrentIndex(
index)
# Aggregation Layer = Entire Area
for index in range(0, aggregation_layer_count):
layer = \
self.impact_merge_dialog.aggregation_layer.itemData(
index, QtCore.Qt.UserRole)
if layer is None:
self.impact_merge_dialog.aggregation_layer.\
setCurrentIndex(index)
# Set Output Directory
#noinspection PyUnresolvedReferences
self.impact_merge_dialog.output_directory.setText(
os.path.join(TEST_DATA_DIR, 'test-impact-merge', 'entire'))
else:
self.impact_merge_dialog.entire_area_mode = False
# Set the current Index of the combobox
for index in range(0, impact_layer_count):
first_combobox = self.impact_merge_dialog.first_layer
layer_name = first_combobox.itemText(index)
if ('population' in layer_name) and ('district' in layer_name):
self.impact_merge_dialog.first_layer.setCurrentIndex(index)
for index in range(0, impact_layer_count):
second_combobox = self.impact_merge_dialog.second_layer
layer_name = second_combobox.itemText(index)
if ('building' in layer_name) and ('district' in layer_name):
self.impact_merge_dialog.second_layer.setCurrentIndex(
index)
# Aggregation Layer = District Jakarta
for index in range(0, aggregation_layer_count):
layer = \
self.impact_merge_dialog.aggregation_layer.itemData(
index, QtCore.Qt.UserRole)
if layer is not None:
self. \
impact_merge_dialog.aggregation_layer. \
setCurrentIndex(index)
# Set output directory
#noinspection PyUnresolvedReferences
self.impact_merge_dialog.output_directory.setText(
os.path.join(TEST_DATA_DIR, 'test-impact-merge', 'aggregated'))
def test_get_project_layers(self):
"""Test get_project_layers function."""
# Test the get_project_layers
self.impact_merge_dialog.get_project_layers()
# On self.impact_merge_dialog.first_layer there must be 4 items
first_layer_expected_number = 4
self.assertEqual(first_layer_expected_number,
self.impact_merge_dialog.first_layer.count())
# On self.impact_merge_dialog.second_layer there must be 4 items
second_layer_expected_number = 4
self.assertEqual(second_layer_expected_number,
self.impact_merge_dialog.second_layer.count())
# On self.impact_merge_dialog.aggregation_layer there must be 2 items
aggregation_layer_expected_number = 2
self.assertEqual(aggregation_layer_expected_number,
self.impact_merge_dialog.aggregation_layer.count())
def test_prepare_input(self):
"""Test prepare_input function."""
# NORMAL CASE
# Test Entire Area
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.prepare_input()
# First impact layer should be the population entire
first_layer_name = \
self.impact_merge_dialog.first_impact['layer'].name()
self.assertIn('population', first_layer_name)
self.assertIn('entire', first_layer_name)
# Second impact layer should be the population entire
second_layer_name = \
self.impact_merge_dialog.second_impact['layer'].name()
self.assertIn('buildings', second_layer_name)
self.assertIn('entire', second_layer_name)
# Chosen aggregaton layer must be none
aggregation_layer = self.impact_merge_dialog.aggregation['layer']
self.assertIsNone(aggregation_layer)
# NORMAL CASE
# Test Aggregated
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
# First impact layer should be the population entire
first_layer_name = \
self.impact_merge_dialog.first_impact['layer'].name()
self.assertIn('population', first_layer_name)
self.assertIn('district', first_layer_name)
# Second impact layer should be the population entire
second_layer_name = \
self.impact_merge_dialog.second_impact['layer'].name()
self.assertIn('buildings', second_layer_name)
self.assertIn('district', second_layer_name)
# Chosen aggregaton layer must be not none
aggregation_layer_name = \
self.impact_merge_dialog.aggregation['layer'].name()
self.assertEqual('district_osm_jakarta', aggregation_layer_name)
# FALL CASE
# First layer combobox index < 0
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.first_layer.setCurrentIndex(-1)
self.assertRaises(InvalidLayerError,
self.impact_merge_dialog.prepare_input)
# FALL CASE
# Second layer combobox index < 0
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.second_layer.setCurrentIndex(-1)
self.assertRaises(InvalidLayerError,
self.impact_merge_dialog.prepare_input)
# FALL CASE
# First impact layer = second impact layer
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.first_layer.setCurrentIndex(1)
self.impact_merge_dialog.second_layer.setCurrentIndex(1)
self.assertRaises(InvalidLayerError,
self.impact_merge_dialog.prepare_input)
def test_require_directory(self):
"""Test require_directory function."""
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.require_directory()
def test_validate_all_layers(self):
"""Test validate_all_layers function."""
# NORMAL CASE
# Test Entire Area mode
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
first_postprocessing_report = \
self.impact_merge_dialog.first_impact['postprocessing_report']
self.assertIn('Detailed gender report', first_postprocessing_report)
second_postprocessing_report = \
self.impact_merge_dialog.second_impact['postprocessing_report']
self.assertIn('Detailed building type report',
second_postprocessing_report)
self.assertEqual(
None,
self.impact_merge_dialog.aggregation['aggregation_attribute'])
# NORMAL CASE
# Test Aggregated Area mode
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
first_postprocessing_report = \
self.impact_merge_dialog.first_impact['postprocessing_report']
self.assertIn('Detailed gender report', first_postprocessing_report)
second_postprocessing_report = \
self.impact_merge_dialog.second_impact['postprocessing_report']
self.assertIn('Detailed building type report',
second_postprocessing_report)
self.assertEqual(
'KAB_NAME',
self.impact_merge_dialog.aggregation['aggregation_attribute'])
# FALL CASE
# There is no keyword post_processing in first_impact_layer
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.prepare_input()
# Change first impact layer to aggregation layer that doesn't have the
# keywords
self.impact_merge_dialog.first_impact['layer'] = \
self.district_jakarta_layer
self.assertRaises(KeywordNotFoundError,
self.impact_merge_dialog.validate_all_layers)
# FALL CASE
# There is no keyword post_processing in second_impact_layer
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.prepare_input()
# Change second impact layer to aggregation layer that doesn't have the
# keywords
self.impact_merge_dialog.second_impact['layer'] = \
self.district_jakarta_layer
self.assertRaises(KeywordNotFoundError,
self.impact_merge_dialog.validate_all_layers)
# FALL CASE
# There is no keyword 'aggregation attribute' in aggregation layer
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
# Change aggregation layer to impact layer that doesn't have the
# keywords
self.impact_merge_dialog.aggregation['layer'] = \
self.population_district_jakarta_layer
self.assertRaises(KeywordNotFoundError,
self.impact_merge_dialog.validate_all_layers)
def test_merge(self):
"""Test merge function."""
# Test Entire Area merged
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
self.impact_merge_dialog.merge()
# There should be 1 pdf files in self.impact_merge_dialog.out_dir
report_list = glob(
os.path.join(self.impact_merge_dialog.out_dir, '*.pdf'))
expected_reports_number = 1
self.assertEqual(len(report_list), expected_reports_number)
# Test Aggregated Area merged
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
self.impact_merge_dialog.merge()
# There should be 3 pdf files in self.impact_merge_dialog.out_dir
report_list = glob(
os.path.join(self.impact_merge_dialog.out_dir, '*.pdf'))
expected_reports_number = 3
self.assertEqual(len(report_list), expected_reports_number)
def test_generate_report_dictionary_from_dom(self):
"""Test generate_report_dictionary_from_dom function."""
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
# Create the DOM
first_postprocessing_report = \
self.impact_merge_dialog.first_impact['postprocessing_report']
second_postprocessing_report = \
self.impact_merge_dialog.second_impact['postprocessing_report']
first_report = ('<body>' + first_postprocessing_report + '</body>')
second_report = ('<body>' + second_postprocessing_report + '</body>')
# Now create a dom document for each
first_document = minidom.parseString(first_report)
second_document = minidom.parseString(second_report)
tables = first_document.getElementsByTagName('table')
tables += second_document.getElementsByTagName('table')
report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
tables)
# There should be 4 keys in that dict
# (3 for each aggregation unit and 1 for total in aggregation unit)
expected_number_of_keys = 4
self.assertEqual(len(report_dict), expected_number_of_keys)
def test_generate_report_summary(self):
"""Test generate_report_summary function."""
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
first_postprocessing_report = \
self.impact_merge_dialog.first_impact['postprocessing_report']
second_postprocessing_report = \
self.impact_merge_dialog.second_impact['postprocessing_report']
first_report = ('<body>' + first_postprocessing_report + '</body>')
second_report = ('<body>' + second_postprocessing_report + '</body>')
# Now create a dom document for each
first_document = minidom.parseString(first_report)
second_document = minidom.parseString(second_report)
first_impact_tables = first_document.getElementsByTagName('table')
second_impact_tables = second_document.getElementsByTagName('table')
first_report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
first_impact_tables)
second_report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
second_impact_tables)
self.impact_merge_dialog.generate_report_summary(
first_report_dict, second_report_dict)
# There should be 4 keys in that dict
# (3 for each aggregation unit and 1 for total in aggregation unit)
expected_number_of_keys = 4
self.assertEqual(len(self.impact_merge_dialog.summary_report),
expected_number_of_keys)
def test_generate_html_reports(self):
"""Test generate_html_reports function."""
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
first_postprocessing_report = \
self.impact_merge_dialog.first_impact['postprocessing_report']
second_postprocessing_report = \
self.impact_merge_dialog.second_impact['postprocessing_report']
first_report = ('<body>' + first_postprocessing_report + '</body>')
second_report = ('<body>' + second_postprocessing_report + '</body>')
# Now create a dom document for each
first_document = minidom.parseString(first_report)
second_document = minidom.parseString(second_report)
first_impact_tables = first_document.getElementsByTagName('table')
second_impact_tables = second_document.getElementsByTagName('table')
first_report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
first_impact_tables)
second_report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
second_impact_tables)
self.impact_merge_dialog.generate_html_reports(first_report_dict,
second_report_dict)
# There should be 4 HTML files generated on output directory
html_list = glob(
os.path.join(self.impact_merge_dialog.out_dir, '*.html'))
expected_html_number = 4
self.assertEqual(len(html_list), expected_html_number)
def test_generate_reports(self):
"""Test generate_reports function."""
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
# Create the DOM
first_postprocessing_report = \
self.impact_merge_dialog.first_impact['postprocessing_report']
second_postprocessing_report = \
self.impact_merge_dialog.second_impact['postprocessing_report']
first_report = ('<body>' + first_postprocessing_report + '</body>')
second_report = ('<body>' + second_postprocessing_report + '</body>')
# Now create a dom document for each
first_document = minidom.parseString(first_report)
second_document = minidom.parseString(second_report)
first_impact_tables = first_document.getElementsByTagName('table')
second_impact_tables = second_document.getElementsByTagName('table')
first_report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
first_impact_tables)
second_report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
second_impact_tables)
self.impact_merge_dialog.generate_report_summary(
first_report_dict, second_report_dict)
self.impact_merge_dialog.generate_html_reports(first_report_dict,
second_report_dict)
# Generate PDF Reports
self.impact_merge_dialog.generate_reports()
# There should be 3 pdf files in self.impact_merge_dialog.out_dir
report_list = glob(
os.path.join(self.impact_merge_dialog.out_dir, '*.pdf'))
expected_reports_number = 3
self.assertEqual(len(report_list), expected_reports_number)
def test_load_template(self):
"""Test load_template function."""
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
# Setup Map Renderer and set all the layer
renderer = QgsMapRenderer()
layer_set = [
self.impact_merge_dialog.first_impact['layer'].id(),
self.impact_merge_dialog.second_impact['layer'].id()
]
# If aggregated, append chosen aggregation layer
if not self.impact_merge_dialog.entire_area_mode:
layer_set.append(
self.impact_merge_dialog.aggregation['layer'].id())
# Set Layer set to renderer
renderer.setLayerSet(layer_set)
# NORMAL CASE: It can find the template
# Create composition
composition = self.impact_merge_dialog.load_template(renderer)
# The type of this composition must be QgsComposition
self.assertEqual(type(composition), QgsComposition)
#FALL CASE: It cannot find the template
self.impact_merge_dialog.template_path = '/it/will/fail/tmp.qpt'
expected_message = 'Error loading template %s' % \
self.impact_merge_dialog.template_path
with self.assertRaises(ReportCreationError) as context:
self.impact_merge_dialog.load_template(renderer)
self.assertEqual(context.exception.message, expected_message)
class ImpactMergeDialogTest(unittest.TestCase):
"""Test Impact Merge Dialog widget."""
#noinspection PyPep8Naming
def setUp(self):
"""Runs before each test."""
#noinspection PyArgumentList
self.map_layer_registry = QgsMapLayerRegistry.instance()
self.register_layers()
# Create Impact Merge Dialog
self.impact_merge_dialog = ImpactMergeDialog(PARENT, IFACE)
# Create test dir
#noinspection PyUnresolvedReferences
test_impact_merge_dir = os.path.join(
TEST_DATA_DIR, 'test-impact-merge')
if not os.path.exists(test_impact_merge_dir):
os.makedirs(test_impact_merge_dir)
# Create test dir for aggregated
#noinspection PyUnresolvedReferences
test_aggregated_dir = os.path.join(
test_impact_merge_dir, 'aggregated')
if not os.path.exists(test_aggregated_dir):
os.makedirs(test_aggregated_dir)
# Create test dir for entire
test_entire_dir = os.path.join(
test_impact_merge_dir, 'entire')
if not os.path.exists(test_entire_dir):
os.makedirs(test_entire_dir)
#noinspection PyPep8Naming
def tearDown(self):
"""Runs after each test."""
# Remove Map Layers
if len(self.map_layer_registry.mapLayers().values()) > 0:
self.map_layer_registry.removeAllMapLayers()
# Delete test dir
#noinspection PyUnresolvedReferences
test_impact_merge_dir = os.path.join(
TEST_DATA_DIR, 'test-impact-merge')
shutil.rmtree(test_impact_merge_dir)
def register_layers(self):
"""Register needed layers to QgsMapLayerRegistry."""
# Register 4 impact layers and aggregation layer
self.population_entire_jakarta_layer, _ = load_layer(
population_entire_jakarta_impact_path,
directory=None)
self.building_entire_jakarta_layer, _ = load_layer(
building_entire_jakarta_impact_path,
directory=None)
self.population_district_jakarta_layer, _ = load_layer(
population_district_jakarta_impact_path,
directory=None)
self.building_district_jakarta_layer, _ = load_layer(
building_district_jakarta_impact_path,
directory=None)
self.district_jakarta_layer, _ = load_layer(
district_jakarta_boundary_path,
directory=None)
layer_list = [self.population_entire_jakarta_layer,
self.population_district_jakarta_layer,
self.building_entire_jakarta_layer,
self.building_district_jakarta_layer,
self.district_jakarta_layer]
# noinspection PyArgumentList
self.map_layer_registry.addMapLayers(layer_list)
def mock_the_dialog(self, test_entire_mode):
impact_layer_count = self.impact_merge_dialog.first_layer.count()
aggregation_layer_count = \
self.impact_merge_dialog.aggregation_layer.count()
if test_entire_mode:
# First impact layer = population entire
# Second impact layer = buildings entire
self.impact_merge_dialog.entire_area_mode = True
# Set the current Index of the combobox
for index in range(0, impact_layer_count):
first_combobox = self.impact_merge_dialog.first_layer
layer_name = first_combobox.itemText(index)
if ('population' in layer_name) and ('entire' in layer_name):
self.impact_merge_dialog.first_layer.setCurrentIndex(
index)
for index in range(0, impact_layer_count):
second_combobox = self.impact_merge_dialog.second_layer
layer_name = second_combobox.itemText(index)
if ('buildings' in layer_name) and ('entire' in layer_name):
self.impact_merge_dialog.second_layer.setCurrentIndex(
index)
# Aggregation Layer = Entire Area
for index in range(0, aggregation_layer_count):
layer = \
self.impact_merge_dialog.aggregation_layer.itemData(
index, QtCore.Qt.UserRole)
if layer is None:
self.impact_merge_dialog.aggregation_layer.\
setCurrentIndex(index)
# Set Output Directory
#noinspection PyUnresolvedReferences
self.impact_merge_dialog.output_directory.setText(
os.path.join(
TEST_DATA_DIR, 'test-impact-merge', 'entire'))
else:
self.impact_merge_dialog.entire_area_mode = False
# Set the current Index of the combobox
for index in range(0, impact_layer_count):
first_combobox = self.impact_merge_dialog.first_layer
layer_name = first_combobox.itemText(index)
if ('population' in layer_name) and ('district' in layer_name):
self.impact_merge_dialog.first_layer.setCurrentIndex(index)
for index in range(0, impact_layer_count):
second_combobox = self.impact_merge_dialog.second_layer
layer_name = second_combobox.itemText(index)
if ('building' in layer_name) and ('district' in layer_name):
self.impact_merge_dialog.second_layer.setCurrentIndex(
index)
# Aggregation Layer = District Jakarta
for index in range(0, aggregation_layer_count):
layer = \
self.impact_merge_dialog.aggregation_layer.itemData(
index, QtCore.Qt.UserRole)
if layer is not None:
self. \
impact_merge_dialog.aggregation_layer. \
setCurrentIndex(index)
# Set output directory
#noinspection PyUnresolvedReferences
self.impact_merge_dialog.output_directory.setText(
os.path.join(
TEST_DATA_DIR, 'test-impact-merge', 'aggregated'))
def test_get_project_layers(self):
"""Test get_project_layers function."""
# Test the get_project_layers
self.impact_merge_dialog.get_project_layers()
# On self.impact_merge_dialog.first_layer there must be 4 items
first_layer_expected_number = 4
self.assertEqual(
first_layer_expected_number,
self.impact_merge_dialog.first_layer.count())
# On self.impact_merge_dialog.second_layer there must be 4 items
second_layer_expected_number = 4
self.assertEqual(
second_layer_expected_number,
self.impact_merge_dialog.second_layer.count())
# On self.impact_merge_dialog.aggregation_layer there must be 2 items
aggregation_layer_expected_number = 2
self.assertEqual(
aggregation_layer_expected_number,
self.impact_merge_dialog.aggregation_layer.count())
def test_prepare_input(self):
"""Test prepare_input function."""
# NORMAL CASE
# Test Entire Area
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.prepare_input()
# First impact layer should be the population entire
first_layer_name = \
self.impact_merge_dialog.first_impact['layer'].name()
self.assertIn('population', first_layer_name)
self.assertIn('entire', first_layer_name)
# Second impact layer should be the population entire
second_layer_name = \
self.impact_merge_dialog.second_impact['layer'].name()
self.assertIn('buildings', second_layer_name)
self.assertIn('entire', second_layer_name)
# Chosen aggregaton layer must be none
aggregation_layer = self.impact_merge_dialog.aggregation['layer']
self.assertIsNone(aggregation_layer)
# NORMAL CASE
# Test Aggregated
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
# First impact layer should be the population entire
first_layer_name = \
self.impact_merge_dialog.first_impact['layer'].name()
self.assertIn('population', first_layer_name)
self.assertIn('district', first_layer_name)
# Second impact layer should be the population entire
second_layer_name = \
self.impact_merge_dialog.second_impact['layer'].name()
self.assertIn('buildings', second_layer_name)
self.assertIn('district', second_layer_name)
# Chosen aggregaton layer must be not none
aggregation_layer_name = \
self.impact_merge_dialog.aggregation['layer'].name()
self.assertEqual('district_osm_jakarta',
aggregation_layer_name)
# FALL CASE
# First layer combobox index < 0
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.first_layer.setCurrentIndex(-1)
self.assertRaises(
InvalidLayerError,
self.impact_merge_dialog.prepare_input)
# FALL CASE
# Second layer combobox index < 0
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.second_layer.setCurrentIndex(-1)
self.assertRaises(
InvalidLayerError,
self.impact_merge_dialog.prepare_input)
# FALL CASE
# First impact layer = second impact layer
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.first_layer.setCurrentIndex(1)
self.impact_merge_dialog.second_layer.setCurrentIndex(1)
self.assertRaises(
InvalidLayerError,
self.impact_merge_dialog.prepare_input)
def test_require_directory(self):
"""Test require_directory function."""
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.require_directory()
def test_validate_all_layers(self):
"""Test validate_all_layers function."""
# NORMAL CASE
# Test Entire Area mode
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
first_postprocessing_report = \
self.impact_merge_dialog.first_impact['postprocessing_report']
self.assertIn(
'Detailed gender report',
first_postprocessing_report)
second_postprocessing_report = \
self.impact_merge_dialog.second_impact['postprocessing_report']
self.assertIn(
'Detailed building type report',
second_postprocessing_report)
self.assertEqual(
None,
self.impact_merge_dialog.aggregation['aggregation_attribute'])
# NORMAL CASE
# Test Aggregated Area mode
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
first_postprocessing_report = \
self.impact_merge_dialog.first_impact['postprocessing_report']
self.assertIn(
'Detailed gender report',
first_postprocessing_report)
second_postprocessing_report = \
self.impact_merge_dialog.second_impact['postprocessing_report']
self.assertIn(
'Detailed building type report',
second_postprocessing_report)
self.assertEqual(
'KAB_NAME',
self.impact_merge_dialog.aggregation['aggregation_attribute'])
# FALL CASE
# There is no keyword post_processing in first_impact_layer
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.prepare_input()
# Change first impact layer to aggregation layer that doesn't have the
# keywords
self.impact_merge_dialog.first_impact['layer'] = \
self.district_jakarta_layer
self.assertRaises(
KeywordNotFoundError,
self.impact_merge_dialog.validate_all_layers)
# FALL CASE
# There is no keyword post_processing in second_impact_layer
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.prepare_input()
# Change second impact layer to aggregation layer that doesn't have the
# keywords
self.impact_merge_dialog.second_impact['layer'] = \
self.district_jakarta_layer
self.assertRaises(
KeywordNotFoundError,
self.impact_merge_dialog.validate_all_layers)
# FALL CASE
# There is no keyword 'aggregation attribute' in aggregation layer
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
# Change aggregation layer to impact layer that doesn't have the
# keywords
self.impact_merge_dialog.aggregation['layer'] = \
self.population_district_jakarta_layer
self.assertRaises(
KeywordNotFoundError,
self.impact_merge_dialog.validate_all_layers)
def test_merge(self):
"""Test merge function."""
# Test Entire Area merged
self.mock_the_dialog(test_entire_mode=True)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
self.impact_merge_dialog.merge()
# There should be 1 pdf files in self.impact_merge_dialog.out_dir
report_list = glob(
os.path.join(
self.impact_merge_dialog.out_dir,
'*.pdf'))
expected_reports_number = 1
self.assertEqual(len(report_list), expected_reports_number)
# Test Aggregated Area merged
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
self.impact_merge_dialog.merge()
# There should be 3 pdf files in self.impact_merge_dialog.out_dir
report_list = glob(
os.path.join(
self.impact_merge_dialog.out_dir,
'*.pdf'))
expected_reports_number = 3
self.assertEqual(len(report_list), expected_reports_number)
def test_generate_report_dictionary_from_dom(self):
"""Test generate_report_dictionary_from_dom function."""
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
# Create the DOM
first_postprocessing_report = \
self.impact_merge_dialog.first_impact['postprocessing_report']
second_postprocessing_report = \
self.impact_merge_dialog.second_impact['postprocessing_report']
first_report = (
'<body>' +
first_postprocessing_report +
'</body>')
second_report = (
'<body>' +
second_postprocessing_report +
'</body>')
# Now create a dom document for each
first_document = minidom.parseString(first_report)
second_document = minidom.parseString(second_report)
tables = first_document.getElementsByTagName('table')
tables += second_document.getElementsByTagName('table')
report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
tables)
# There should be 4 keys in that dict
# (3 for each aggregation unit and 1 for total in aggregation unit)
expected_number_of_keys = 4
self.assertEqual(len(report_dict), expected_number_of_keys)
def test_generate_report_summary(self):
"""Test generate_report_summary function."""
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
first_postprocessing_report = \
self.impact_merge_dialog.first_impact['postprocessing_report']
second_postprocessing_report = \
self.impact_merge_dialog.second_impact['postprocessing_report']
first_report = (
'<body>' +
first_postprocessing_report +
'</body>')
second_report = (
'<body>' +
second_postprocessing_report +
'</body>')
# Now create a dom document for each
first_document = minidom.parseString(first_report)
second_document = minidom.parseString(second_report)
first_impact_tables = first_document.getElementsByTagName('table')
second_impact_tables = second_document.getElementsByTagName('table')
first_report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
first_impact_tables)
second_report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
second_impact_tables)
self.impact_merge_dialog.generate_report_summary(
first_report_dict, second_report_dict)
# There should be 4 keys in that dict
# (3 for each aggregation unit and 1 for total in aggregation unit)
expected_number_of_keys = 4
self.assertEqual(len(self.impact_merge_dialog.summary_report),
expected_number_of_keys)
def test_generate_html_reports(self):
"""Test generate_html_reports function."""
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
first_postprocessing_report = \
self.impact_merge_dialog.first_impact['postprocessing_report']
second_postprocessing_report = \
self.impact_merge_dialog.second_impact['postprocessing_report']
first_report = (
'<body>' +
first_postprocessing_report +
'</body>')
second_report = (
'<body>' +
second_postprocessing_report +
'</body>')
# Now create a dom document for each
first_document = minidom.parseString(first_report)
second_document = minidom.parseString(second_report)
first_impact_tables = first_document.getElementsByTagName('table')
second_impact_tables = second_document.getElementsByTagName('table')
first_report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
first_impact_tables)
second_report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
second_impact_tables)
self.impact_merge_dialog.generate_html_reports(
first_report_dict, second_report_dict)
# There should be 4 HTML files generated on output directory
html_list = glob(
os.path.join(
self.impact_merge_dialog.out_dir,
'*.html'))
expected_html_number = 4
self.assertEqual(len(html_list), expected_html_number)
def test_generate_reports(self):
"""Test generate_reports function."""
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
# Create the DOM
first_postprocessing_report = \
self.impact_merge_dialog.first_impact['postprocessing_report']
second_postprocessing_report = \
self.impact_merge_dialog.second_impact['postprocessing_report']
first_report = (
'<body>' +
first_postprocessing_report +
'</body>')
second_report = (
'<body>' +
second_postprocessing_report +
'</body>')
# Now create a dom document for each
first_document = minidom.parseString(first_report)
second_document = minidom.parseString(second_report)
first_impact_tables = first_document.getElementsByTagName('table')
second_impact_tables = second_document.getElementsByTagName('table')
first_report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
first_impact_tables)
second_report_dict = \
self.impact_merge_dialog.generate_report_dictionary_from_dom(
second_impact_tables)
self.impact_merge_dialog.generate_report_summary(
first_report_dict, second_report_dict)
self.impact_merge_dialog.generate_html_reports(
first_report_dict, second_report_dict)
# Generate PDF Reports
self.impact_merge_dialog.generate_reports()
# There should be 3 pdf files in self.impact_merge_dialog.out_dir
report_list = glob(
os.path.join(
self.impact_merge_dialog.out_dir,
'*.pdf'))
expected_reports_number = 3
self.assertEqual(len(report_list), expected_reports_number)
def test_load_template(self):
"""Test load_template function."""
self.mock_the_dialog(test_entire_mode=False)
self.impact_merge_dialog.prepare_input()
self.impact_merge_dialog.validate_all_layers()
# Setup Map Renderer and set all the layer
renderer = QgsMapRenderer()
layer_set = [self.impact_merge_dialog.first_impact['layer'].id(),
self.impact_merge_dialog.second_impact['layer'].id()]
# If aggregated, append chosen aggregation layer
if not self.impact_merge_dialog.entire_area_mode:
layer_set.append(
self.impact_merge_dialog.aggregation['layer'].id())
# Set Layer set to renderer
renderer.setLayerSet(layer_set)
# NORMAL CASE: It can find the template
# Create composition
composition = self.impact_merge_dialog.load_template(renderer)
# The type of this composition must be QgsComposition
self.assertEqual(type(composition), QgsComposition)
#FALL CASE: It cannot find the template
self.impact_merge_dialog.template_path = '/it/will/fail/tmp.qpt'
expected_message = 'Error loading template %s' % \
self.impact_merge_dialog.template_path
with self.assertRaises(ReportCreationError) as context:
self.impact_merge_dialog.load_template(renderer)
self.assertEqual(context.exception.message, expected_message)