def test_validateTraining_TrainWindow(qtbot):
widget = TrainWindow()
qtbot.addWidget(widget)
widget._precision_level = 2
widget.validateTraining()
assert widget.buttonConnected is False
widget._is_model_ready = True
widget.validateTraining()
assert widget.buttonConnected is False
widget._is_dataset_linked = True
widget.validateTraining()
assert widget.buttonConnected is False
widget._is_labellist_linked = True
widget.validateTraining()
assert widget.buttonConnected is False
widget._precision_level = 1
widget.validateTraining()
assert widget.buttonConnected is True
widget.buttonConnected = False
widget._precision_level = 2
widget._is_dataset_labelled = True
widget.validateTraining()
assert widget.buttonConnected is True
def test_validateDataset_TrainWindow(qtbot):
widget = TrainWindow()
qtbot.addWidget(widget)
widget._precision_level = 1
widget._path_to_dataset = "invalid_path_to_dataset"
qtbot.mouseClick(widget.validate_button, QtCore.Qt.LeftButton)
assert widget._is_dataset_labelled is False
widget._precision_level = 2
qtbot.mouseClick(widget.validate_button, QtCore.Qt.LeftButton)
assert widget._is_dataset_labelled is False
widget._precision_level = 3
qtbot.mouseClick(widget.validate_button, QtCore.Qt.LeftButton)
assert widget._is_dataset_labelled is False
def test_populateModelSelector_TrainWindow(qtbot):
widget = TrainWindow()
qtbot.addWidget(widget)
widget._precision_level = 1
widget.populateModelSelector()
assert len(widget._model_list) == 7
widget._precision_level = 2
widget.populateModelSelector()
assert widget._model_list[0] == 'fasterrcnn'
widget._precision_level = 3
widget.populateModelSelector()
assert widget._model_list[0] == 'maskrcnn'
