def test_get_data_fail():
data_under_test = FakeInstrument([get_component()])
under_test = ComponentTreeModel(data_under_test)
under_test.createIndex(0, 0, data_under_test[0])
assert under_test.data(QModelIndex(), Qt.DisplayRole) is None
def test_get_flags_fail():
data_under_test = FakeInstrument([get_component()])
under_test = ComponentTreeModel(data_under_test)
under_test.createIndex(0, 0, data_under_test[0])
assert under_test.flags(QModelIndex()) is Qt.NoItemFlags
def test_component_has_2_rows():
data_under_test = FakeInstrument([get_component()])
under_test = ComponentTreeModel(data_under_test)
test_index = under_test.createIndex(0, 0, data_under_test[0])
assert under_test.rowCount(test_index) == 2
def test_get_flags_component():
data_under_test = FakeInstrument([get_component()])
under_test = ComponentTreeModel(data_under_test)
index = under_test.createIndex(0, 0, data_under_test[0])
assert under_test.flags(index) == (Qt.ItemIsEnabled | Qt.ItemIsSelectable)
def test_get_component_parent():
data_under_test = FakeInstrument([get_component()])
under_test = ComponentTreeModel(data_under_test)
test_index = under_test.createIndex(0, 0, data_under_test[0])
assert under_test.parent(test_index) == QModelIndex()
def test_get_data_success_2():
data_under_test = FakeInstrument([get_component()])
under_test = ComponentTreeModel(data_under_test)
test_index = under_test.createIndex(0, 0, data_under_test[0])
assert under_test.data(test_index, Qt.SizeHintRole) is None
def test_get_flags_component_info():
data_under_test = FakeInstrument([get_component()])
under_test = ComponentTreeModel(data_under_test)
item = ComponentInfo(parent=data_under_test[0])
index = under_test.createIndex(0, 0, item)
assert under_test.flags(index) == Qt.ItemIsEnabled
def test_rowCount_gets_unknown_type():
data_under_test = FakeInstrument()
under_test = ComponentTreeModel(data_under_test)
test_index = under_test.createIndex(0, 0, {})
with pytest.raises(RuntimeError):
under_test.rowCount(test_index)
def test_get_flags_transformation_list():
data_under_test = FakeInstrument([get_component()])
under_test = ComponentTreeModel(data_under_test)
component = data_under_test[0]
component.stored_transforms = component.transforms
index = under_test.createIndex(0, 0, component.stored_transforms)
assert under_test.flags(index) == Qt.ItemIsEnabled | Qt.ItemIsSelectable
def test_get_component_info_parent():
data_under_test = FakeInstrument([get_component()])
under_test = ComponentTreeModel(data_under_test)
# Creating ComponentInfo in-line causes a segmentation error
temp_component_info = ComponentInfo(parent=data_under_test[0])
test_index = under_test.createIndex(0, 0, temp_component_info)
assert under_test.parent(
test_index).internalPointer() is data_under_test[0]
def test_transformation_list_has_0_rows():
data_under_test = FakeInstrument([get_component()])
under_test = ComponentTreeModel(data_under_test)
data_under_test[0].stored_transforms = data_under_test[0].transforms
test_index = under_test.createIndex(0, 0,
data_under_test[0].stored_transforms)
assert under_test.rowCount(test_index) == 0
def test_transformation_link_has_0_rows():
component = get_component()
translation = component.add_translation(QVector3D(1.0, 0.0, 0.0))
component.depends_on = translation
data_under_test = FakeInstrument([component])
component.stored_transforms = component.transforms
under_test = ComponentTreeModel(data_under_test)
test_index = under_test.createIndex(0, 0, component.stored_transforms[0])
assert under_test.rowCount(test_index) == 0
def test_get_flags_other():
data_under_test = FakeInstrument([get_component()])
under_test = ComponentTreeModel(data_under_test)
class TestObject:
pass
test_item = TestObject()
index = under_test.createIndex(0, 0, test_item)
assert (under_test.flags(index) == Qt.ItemIsEnabled | Qt.ItemIsSelectable
| Qt.ItemIsEditable)
def test_get_transform_list_parent():
data_under_test = FakeInstrument([get_component()])
under_test = ComponentTreeModel(data_under_test)
data_under_test[0].stored_transforms = data_under_test[0].transforms
test_index = under_test.createIndex(0, 0,
data_under_test[0].stored_transforms)
temp_parent = under_test.parent(test_index)
assert temp_parent.internalPointer() is data_under_test[0]
assert temp_parent.row() == 0
def test_get_transformation_link_parent():
component = get_component()
data_under_test = FakeInstrument([component])
component.stored_transforms = component.transforms
transform_link = LinkTransformation(component.stored_transforms)
component.stored_transforms.link = transform_link
component.stored_transforms.has_link = True
under_test = ComponentTreeModel(data_under_test)
test_index = under_test.createIndex(0, 0, transform_link)
found_parent = under_test.parent(test_index)
assert found_parent.internalPointer(
) == data_under_test[0].stored_transforms
assert found_parent.row() == 1
def test_get_transformation_parent():
component = get_component()
translation = component.add_translation(QVector3D(1.0, 0.0, 0.0))
component.depends_on = translation
data_under_test = FakeInstrument([component])
component.stored_transforms = component.transforms
translation.parent = component.stored_transforms
under_test = ComponentTreeModel(data_under_test)
test_index = under_test.createIndex(0, 0, translation)
found_parent = under_test.parent(test_index)
assert found_parent.internalPointer(
) == data_under_test[0].stored_transforms
assert found_parent.row() == 1
