def test_column_type_checking_produces_error(self):
model = MappingPreviewModel()
model.reset_model([["Not a valid number", "2.4"], ["1", "3"]])
model.set_type(0, value_to_convert_spec('float'))
error_index = (0, 0)
self.assertEqual(len(model._column_type_errors), 1)
self.assertEqual(model._row_type_errors, {})
self.assertTrue(error_index in model._column_type_errors)
self.assertEqual(model.data(model.index(*error_index)), "Error")
# if we add a pivoted mapping for the row with the error, the error should not be shown
mapping = MappingSpecModel(
dict_to_map({
"map_type": "ObjectClass",
"name": {
"map_type": "row",
"value_reference": 0
}
}))
model.set_mapping(mapping)
self.assertEqual(model.data(model.index(*error_index)),
"Not a valid number")
# or if we add a mapping where there reading starts from a row bellow the error, the error should not be shown.
mapping = MappingSpecModel(
dict_to_map({
"map_type": "ObjectClass",
"read_start_row": 1
}))
model.set_mapping(mapping)
self.assertEqual(model.data(model.index(*error_index)),
"Not a valid number")
def use_settings(self, settings):
self.table_mappings = {
table: MappingListModel([dict_to_map(m) for m in mappings])
for table, mappings in settings.get("table_mappings", {}).items()
}
table_types = {
tn: {
int(col): value_to_convert_spec(spec)
for col, spec in cols.items()
}
for tn, cols in settings.get("table_types", {}).items()
}
table_row_types = {
tn: {
int(col): value_to_convert_spec(spec)
for col, spec in cols.items()
}
for tn, cols in settings.get("table_row_types", {}).items()
}
self.connector.set_table_options(settings.get("table_options", {}))
self.connector.set_table_types(table_types)
self.connector.set_table_row_types(table_row_types)
self.selected_source_tables.update(
set(settings.get("selected_tables", [])))
for i in range(self._ui.source_list.count()):
item = self._ui.source_list.item(i)
selected = Qt.Checked if item.text(
) in self.selected_source_tables else Qt.Unchecked
item.setCheckState(selected)
if self._ui.source_list.selectedItems():
self.select_table(self._ui.source_list.selectedItems()[0])
def test_row_type_checking_produces_error(self):
model = MappingPreviewModel()
model.reset_model([["1", "2.4"], ["Not a valid number", "3"]])
model.set_type(1,
value_to_convert_spec('float'),
orientation=Qt.Vertical)
error_index = (1, 0)
self.assertEqual(len(model._row_type_errors), 1)
self.assertEqual(model._column_type_errors, {})
self.assertTrue(error_index in model._row_type_errors)
# Error should only be shown if we have a pivot mapping on that row.
self.assertEqual(model.data(model.index(*error_index)),
"Not a valid number")
# if we add mapping error should be shown.
mapping = MappingSpecModel(
dict_to_map({
"map_type": "ObjectClass",
"name": {
"map_type": "row",
"value_reference": 1
}
}))
model.set_mapping(mapping)
self.assertEqual(model.data(model.index(*error_index)), "Error")
def test_mapping_column_and_pivot_colors(self):
model = MappingPreviewModel()
model.reset_model([[1, 2], [3, 4]])
# row mapping
mapping = MappingSpecModel(
dict_to_map({
"map_type": "ObjectClass",
"name": 0,
"object": {
"map_type": "row",
"value_reference": 0
}
}))
model.set_mapping(mapping)
# no color showing where row and column mapping intersect
self.assertEqual(
model.data(model.index(0, 0), role=Qt.BackgroundColorRole), None)
self.assertEqual(
model.data(model.index(0, 1), role=Qt.BackgroundColorRole),
_MAPPING_COLORS["entity"])
self.assertEqual(
model.data(model.index(1, 0), role=Qt.BackgroundColorRole),
_MAPPING_COLORS["entity class"])
self.assertEqual(
model.data(model.index(1, 1), role=Qt.BackgroundColorRole), None)
def test_data_when_mapping_relationship_class_without_objects_or_parameters(self):
model = MappingSpecModel(
dict_to_map({"map_type": "RelationshipClass", "name": None, "object_classes": None, "object": None}),
"connector's name",
)
self.assertEqual(model.rowCount(), 3)
self.assertEqual(model.columnCount(), 3)
index = model.index(0, 0)
self.assertEqual(index.data(), "Relationship class names")
index = model.index(1, 0)
self.assertEqual(index.data(), "Object class names 1")
index = model.index(2, 0)
self.assertEqual(index.data(), "Object names 1")
index = model.index(0, 1)
self.assertEqual(index.data(), "None")
index = model.index(1, 1)
self.assertEqual(index.data(), "None")
index = model.index(2, 1)
self.assertEqual(index.data(), "None")
index = model.index(0, 2)
self.assertEqual(index.data(), "")
self.assertEqual(index.data(Qt.BackgroundColorRole), _ERROR_COLOR)
self.assertTrue(index.data(Qt.ToolTipRole))
index = model.index(1, 2)
self.assertEqual(index.data(), "")
self.assertEqual(index.data(Qt.BackgroundColorRole), _ERROR_COLOR)
self.assertTrue(index.data(Qt.ToolTipRole))
index = model.index(2, 2)
self.assertEqual(index.data(), "")
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
def test_data_when_mapping_invalid_object_class_with_parameters(self):
indexed_parameter_mapping_dict = {
"map_type": "parameter",
"name": {'map_type': None},
"parameter_type": "map",
"value": {"map_type": None},
"extra_dimensions": [{"map_type": None}],
}
mapping_dict = {
"map_type": "ObjectClass",
"name": None,
"objects": None,
"parameters": indexed_parameter_mapping_dict,
}
model = MappingSpecModel(dict_to_map(mapping_dict), "connector's name")
self.assertEqual(model.rowCount(), 5)
self.assertEqual(model.columnCount(), 3)
index = model.index(0, 0)
self.assertEqual(index.data(), "Object class names")
index = model.index(1, 0)
self.assertEqual(index.data(), "Object names")
index = model.index(2, 0)
self.assertEqual(index.data(), "Parameter names")
index = model.index(3, 0)
self.assertEqual(index.data(), "Parameter values")
index = model.index(4, 0)
self.assertEqual(index.data(), "Parameter map index")
for row in range(5):
index = model.index(row, 1)
self.assertEqual(index.data(), "None")
index = model.index(row, 2)
self.assertEqual(index.data(), "")
self.assertEqual(index.data(Qt.BackgroundColorRole), _ERROR_COLOR)
self.assertTrue(index.data(Qt.ToolTipRole))
def test_mapping_column_colors(self):
model = MappingPreviewModel()
model.reset_model([[1, 2], [3, 4]])
# column mapping
mapping = MappingSpecModel(
dict_to_map({
"map_type": "ObjectClass",
"name": 0
}))
model.set_mapping(mapping)
self.assertEqual(
model.data(model.index(0, 0), role=Qt.BackgroundColorRole),
_MAPPING_COLORS["entity class"])
self.assertEqual(
model.data(model.index(1, 0), role=Qt.BackgroundColorRole),
_MAPPING_COLORS["entity class"])
# row not showing color if the start reading row is specified
mapping = MappingSpecModel(
dict_to_map({
"map_type": "ObjectClass",
"name": 0,
"read_start_row": 1
}))
model.set_mapping(mapping)
self.assertEqual(
model.data(model.index(0, 0), role=Qt.BackgroundColorRole), None)
self.assertEqual(
model.data(model.index(1, 0), role=Qt.BackgroundColorRole),
_MAPPING_COLORS["entity class"])
# row not showing color if the row is pivoted
mapping = MappingSpecModel(
dict_to_map({
"map_type": "ObjectClass",
"name": 0,
"object": {
"map_type": "row",
"value_reference": 0
}
}))
model.set_mapping(mapping)
self.assertNotEqual(
model.data(model.index(0, 0), role=Qt.BackgroundColorRole),
_MAPPING_COLORS["entity class"])
self.assertEqual(
model.data(model.index(1, 0), role=Qt.BackgroundColorRole),
_MAPPING_COLORS["entity class"])
def test_data_when_mapping_valid_object_class_with_parameters(self):
indexed_parameter_mapping_dict = {
"map_type": "parameter",
"name": {'map_type': 'column', 'reference': 99},
"parameter_type": "map",
"value": {"reference": 23, "map_type": "column"},
"extra_dimensions": [{"reference": "fifth column", "map_type": "column"}],
}
mapping_dict = {
"map_type": "ObjectClass",
"parameters": indexed_parameter_mapping_dict,
"name": {"reference": "class_name", "map_type": "constant"},
"objects": {"reference": "object_name", "map_type": "constant"},
}
model = MappingSpecModel(dict_to_map(mapping_dict), "connector's name")
self.assertEqual(model.rowCount(), 5)
self.assertEqual(model.columnCount(), 3)
index = model.index(0, 0)
self.assertEqual(index.data(), "Object class names")
index = model.index(1, 0)
self.assertEqual(index.data(), "Object names")
index = model.index(2, 0)
self.assertEqual(index.data(), "Parameter names")
index = model.index(3, 0)
self.assertEqual(index.data(), "Parameter values")
index = model.index(4, 0)
self.assertEqual(index.data(), "Parameter map index")
index = model.index(0, 1)
self.assertEqual(index.data(), "Constant")
index = model.index(1, 1)
self.assertEqual(index.data(), "Constant")
index = model.index(2, 1)
self.assertEqual(index.data(), "Column")
index = model.index(3, 1)
self.assertEqual(index.data(), "Column")
index = model.index(4, 1)
self.assertEqual(index.data(), "Column")
index = model.index(0, 2)
self.assertEqual(index.data(), "class_name")
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
index = model.index(1, 2)
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
self.assertEqual(index.data(), "object_name")
index = model.index(2, 2)
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
self.assertEqual(index.data(), 99)
index = model.index(3, 2)
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
self.assertEqual(index.data(), 23)
index = model.index(4, 2)
self.assertEqual(index.data(), "fifth column")
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
def test_mapping_pivoted_colors(self):
model = MappingPreviewModel()
model.reset_model([[1, 2], [3, 4]])
# row mapping
mapping = MappingSpecModel(
dict_to_map({
"map_type": "ObjectClass",
"object": {
"map_type": "row",
"value_reference": 0
}
}))
model.set_mapping(mapping)
self.assertEqual(
model.data(model.index(0, 0), role=Qt.BackgroundColorRole),
_MAPPING_COLORS["entity"])
self.assertEqual(
model.data(model.index(0, 1), role=Qt.BackgroundColorRole),
_MAPPING_COLORS["entity"])
self.assertEqual(
model.data(model.index(1, 0), role=Qt.BackgroundColorRole), None)
# column not showing color if the columns is skipped
mapping = MappingSpecModel(
dict_to_map({
"map_type": "ObjectClass",
"object": {
"map_type": "row",
"value_reference": 0
},
"skip_columns": [0]
}))
model.set_mapping(mapping)
self.assertEqual(
model.data(model.index(0, 0), role=Qt.BackgroundColorRole), None)
self.assertEqual(
model.data(model.index(0, 1), role=Qt.BackgroundColorRole),
_MAPPING_COLORS["entity"])
self.assertEqual(
model.data(model.index(1, 0), role=Qt.BackgroundColorRole), None)
def test_data_when_mapping_valid_object_class_with_pivoted_parameters(self):
array_parameter_mapping_dict = {
"map_type": "parameter",
"name": 2,
"parameter_type": "array",
"value": {"map_type": "row", "reference": 0},
}
mapping_dict = {"map_type": "ObjectClass", "name": 0, "objects": 1, "parameters": array_parameter_mapping_dict}
model = MappingSpecModel(dict_to_map(mapping_dict), "connector's name")
self.assertEqual(model.rowCount(), 4)
self.assertEqual(model.columnCount(), 3)
index = model.index(0, 0)
self.assertEqual(index.data(), "Object class names")
index = model.index(1, 0)
self.assertEqual(index.data(), "Object names")
index = model.index(2, 0)
self.assertEqual(index.data(), "Parameter names")
index = model.index(3, 0)
self.assertEqual(index.data(), "Parameter values")
index = model.index(0, 1)
self.assertEqual(index.data(), "Column")
index = model.index(1, 1)
self.assertEqual(index.data(), "Column")
index = model.index(2, 1)
self.assertEqual(index.data(), "Column")
index = model.index(3, 1)
self.assertEqual(index.data(), "Pivoted")
index = model.index(0, 2)
self.assertEqual(index.data(), 0)
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
index = model.index(1, 2)
self.assertEqual(index.data(), 1)
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
index = model.index(2, 2)
self.assertEqual(index.data(), 2)
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
index = model.index(3, 2)
self.assertEqual(index.data(), "Pivoted values")
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
def use_settings(self, settings):
try:
self.table_mappings = {
table: MappingListModel([dict_to_map(m) for m in mappings],
table)
for table, mappings in settings.get("table_mappings",
{}).items()
}
except ValueError as error:
self._ui_error.showMessage(f"{error}")
return
table_types = {
tn: {
int(col): value_to_convert_spec(spec)
for col, spec in cols.items()
}
for tn, cols in settings.get("table_types", {}).items()
}
table_row_types = {
tn: {
int(col): value_to_convert_spec(spec)
for col, spec in cols.items()
}
for tn, cols in settings.get("table_row_types", {}).items()
}
self.connector.set_table_options(settings.get("table_options", {}))
self.connector.set_table_types(table_types)
self.connector.set_table_row_types(table_row_types)
self._ui.source_list.blockSignals(True)
self._ui.source_list.clear()
selected_tables = settings.get("selected_tables")
if selected_tables is None:
selected_tables = set(self.table_mappings.keys())
for table_name in self.table_mappings:
item = QListWidgetItem()
item.setText(table_name)
item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
item.setCheckState(Qt.Checked if table_name in
selected_tables else Qt.Unchecked)
self._ui.source_list.addItem(item)
self._ui.source_list.blockSignals(False)
def test_data_when_mapping_multidimensional_relationship_class_with_parameters(self):
indexed_parameter_mapping_dict = {
"map_type": "parameter",
"name": {'map_type': 'column', 'reference': 99},
"parameter_type": "map",
"value": {"reference": 23, "map_type": "column"},
"extra_dimensions": [{"reference": "fifth column", "map_type": "column"}],
}
mapping_dict = {
"map_type": "RelationshipClass",
"name": {"map_type": "constant", "reference": "relationship class name"},
"object_classes": [
{"map_type": "column", "reference": "column header"},
{"map_type": "constant", "reference": "second class"},
],
"objects": [{"map_type": "column", "reference": 21}, {"map_type": "column", "reference": 22}],
"parameters": indexed_parameter_mapping_dict,
}
model = MappingSpecModel(dict_to_map(mapping_dict), "connector's name")
self.assertEqual(model.rowCount(), 8)
self.assertEqual(model.columnCount(), 3)
index = model.index(0, 0)
self.assertEqual(index.data(), "Relationship class names")
index = model.index(1, 0)
self.assertEqual(index.data(), "Object class names 1")
index = model.index(2, 0)
self.assertEqual(index.data(), "Object class names 2")
index = model.index(3, 0)
self.assertEqual(index.data(), "Object names 1")
index = model.index(4, 0)
self.assertEqual(index.data(), "Object names 2")
index = model.index(5, 0)
self.assertEqual(index.data(), "Parameter names")
index = model.index(6, 0)
self.assertEqual(index.data(), "Parameter values")
index = model.index(7, 0)
self.assertEqual(index.data(), "Parameter map index")
index = model.index(0, 1)
self.assertEqual(index.data(), "Constant")
index = model.index(1, 1)
self.assertEqual(index.data(), "Column")
index = model.index(2, 1)
self.assertEqual(index.data(), "Constant")
index = model.index(3, 1)
self.assertEqual(index.data(), "Column")
index = model.index(4, 1)
self.assertEqual(index.data(), "Column")
index = model.index(5, 1)
self.assertEqual(index.data(), "Column")
index = model.index(6, 1)
self.assertEqual(index.data(), "Column")
index = model.index(7, 1)
self.assertEqual(index.data(), "Column")
index = model.index(0, 2)
self.assertEqual(index.data(), "relationship class name")
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
index = model.index(1, 2)
self.assertEqual(index.data(), "column header")
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
index = model.index(2, 2)
self.assertEqual(index.data(), "second class")
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
index = model.index(3, 2)
self.assertEqual(index.data(), 21)
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
index = model.index(4, 2)
self.assertEqual(index.data(), 22)
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
index = model.index(5, 2)
self.assertEqual(index.data(), 99)
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
index = model.index(6, 2)
self.assertEqual(index.data(), 23)
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))
index = model.index(7, 2)
self.assertEqual(index.data(), "fifth column")
self.assertEqual(index.data(Qt.BackgroundColorRole), None)
self.assertFalse(index.data(Qt.ToolTipRole))