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_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_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_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_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)