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'