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