def test_addRemoveBits(qtbot, open_window, good_data): table_model = DigitalStateTableModel(allow_is_used=True) table_model.setHalPins(good_data[0]) table_model.setStates(good_data[1]) table_model.setIsUsed(good_data[2]) table_view = open_window(DigitalStateTableView) table_view.setWindowTitle("Digital State Table View") table_view.setModel(table_model) table_view.resize(table_view.sizeHint() * 2) assert table_view.isVisible() starting_rows = table_model.rowCount() #Add One table_view.addBit() qtbot.wait(TestingFlags.TEST_WAIT_SHORT) assert starting_rows << 1 == table_model.rowCount() #Remove One table_view.removeBit() qtbot.wait(TestingFlags.TEST_WAIT_SHORT) assert starting_rows == table_model.rowCount() #Remove Another table_view.removeBit() qtbot.wait(TestingFlags.TEST_WAIT_SHORT) assert starting_rows >> 1 == table_model.rowCount() #Remove One Too Many! with pytest.raises(ValueError): table_view.removeBit() qtbot.wait(TestingFlags.TEST_WAIT_SHORT) #Add One table_view.addBit() qtbot.wait(TestingFlags.TEST_WAIT_SHORT) assert starting_rows == table_model.rowCount() table_model.setAllowedHalPins( ['None', 'pin_1', 'pin_2', 'pin_33', 'pin_4']) table_view.setHalPin(0, 'pin_1') table_view.setHalPin(1, 'pin_2') menu = table_view.generateContextMenu() assert isinstance(menu, QtWidgets.QMenu) if TestingFlags.ENABLE_MANUAL_TESTING: MessageBox( "Digital State Table Manual Testing", "Test that the GUI Name and is Used can be edited\n Row 0 is used must be True\n Right click add and remove bits" ) qtbot.stopForInteraction()
def test_data(good_states): for data in good_states: table_model = DigitalStateTableModel(allow_is_used=True) table_model.setHalPins(data[0]) table_model.setStates(data[1]) table_model.setIsUsed(data[2]) for i, itm in enumerate(data[1]): assert itm == table_model.data( table_model.index(i, table_model.nameColumn(), QtCore.QModelIndex()), QtCore.Qt.DisplayRole) for i, itm in enumerate(data[2]): assert itm == table_model.data( table_model.index(i, table_model.isUsedColumn(), QtCore.QModelIndex()), QtCore.Qt.DisplayRole)
def test_headerData(good_states): for data in good_states: table_model = DigitalStateTableModel(allow_is_used=True) table_model.setHalPins(data[0]) table_model.setStates(data[1]) table_model.setIsUsed(data[2]) assert 'GUI Name' == table_model.headerData(table_model.nameColumn(), QtCore.Qt.Horizontal, QtCore.Qt.DisplayRole) assert 'Is Used' == table_model.headerData(table_model.isUsedColumn(), QtCore.Qt.Horizontal, QtCore.Qt.DisplayRole) assert 'Is Used' != table_model.headerData(0, QtCore.Qt.Horizontal, QtCore.Qt.DisplayRole) #check vertical header num_rows = len(data[1]) for row in range(num_rows): assert str(row) == table_model.headerData(row, QtCore.Qt.Vertical, QtCore.Qt.DisplayRole)
def test_setDataArray(good_states): for data in good_states: table_model = DigitalStateTableModel(allow_is_used=True) table_model.setHalPins(data[0]) table_model.setStates(data[1]) table_model.setIsUsed(data[2]) assert data[0] == table_model.halPins() assert data[1] == table_model.states() assert data[2] == table_model.isUsed() table_model.setHalPins(['pin_a']) with pytest.raises(ValueError): data = ['btn_1', 'btn_2', 'btn_1_too_many'] table_model.setStates(data) with pytest.raises(TypeError): data = [True, 'false'] table_model.setIsUsed(data) with pytest.raises(TypeError): data = [("BTN 1", True), ("BTN 4", True)] table_model.setStates(data)
def test_setData(good_states): for data in good_states: table_model = DigitalStateTableModel(allow_is_used=True) table_model.setHalPins(data[0]) table_model.setStates(data[1]) table_model.setIsUsed(data[2]) is_used_col = table_model.isUsedColumn() name_col = table_model.nameColumn() num_bits = table_model.numberOfBits() num_rows = len(data[1]) #Try setting some of the isUsed column table_model.setData( table_model.index(0, is_used_col, QtCore.QModelIndex()), False, QtCore.Qt.EditRole) val = table_model.data( table_model.index(0, is_used_col, QtCore.QModelIndex()), QtCore.Qt.DisplayRole) assert val == True table_model.setData( table_model.index(num_rows - 1, is_used_col, QtCore.QModelIndex()), False, QtCore.Qt.EditRole) val = table_model.data( table_model.index(num_rows - 1, is_used_col, QtCore.QModelIndex()), QtCore.Qt.DisplayRole) assert val == False table_model.setData( table_model.index(0, is_used_col, QtCore.QModelIndex()), True, QtCore.Qt.EditRole) val = table_model.data( table_model.index(0, is_used_col, QtCore.QModelIndex()), QtCore.Qt.DisplayRole) assert val == True table_model.setData( table_model.index(num_rows - 1, is_used_col, QtCore.QModelIndex()), True, QtCore.Qt.EditRole) val = table_model.data( table_model.index(num_rows - 1, is_used_col, QtCore.QModelIndex()), QtCore.Qt.DisplayRole) assert val == True #Try setting some of the name column table_model.setData( table_model.index(0, name_col, QtCore.QModelIndex()), "Fast Off", QtCore.Qt.EditRole) val = table_model.data( table_model.index(0, name_col, QtCore.QModelIndex()), QtCore.Qt.DisplayRole) assert val == "Fast Off" table_model.setData( table_model.index(num_rows - 1, name_col, QtCore.QModelIndex()), "Start", QtCore.Qt.EditRole) val = table_model.data( table_model.index(num_rows - 1, name_col, QtCore.QModelIndex()), QtCore.Qt.DisplayRole) assert val == "Start" #Try setting 0,0 to True, this should always be false table_model.setData(table_model.index(0, 0, QtCore.QModelIndex()), True, QtCore.Qt.EditRole) val = table_model.data(table_model.index(0, 0, QtCore.QModelIndex()), QtCore.Qt.DisplayRole) assert val == False #Try setting 0,0 to True, this should always be false table_model.setData( table_model.index(0, num_bits - 1, QtCore.QModelIndex()), False, QtCore.Qt.EditRole) val = table_model.data( table_model.index(0, num_bits - 1, QtCore.QModelIndex()), QtCore.Qt.DisplayRole) assert val == False