pytest-bdd-debug

gistfile1.txt

@when('I start to train')
def start_train(tmpdir, mocker, context, qtbot, monkeypatch):
    
    model_folder = os.path.join(tmpdir, 'model_folder')
    mocker.patch.object(window.main_view.dialog_train, 'model_folder', return_value=(model_folder, None))

    # click the train
    window.main_view._ui.actionTrain.trigger()

    # begin to train
    # import pdb; pdb.set_trace()
    window.main_view.dialog_train._ui.pushbutton_step1_next.click()
    # on_train_start
        # with qtbot.waitSignal(window.main_view.dialog_train.timer.timeout.on_train_start()):
    out_folder_train = window.main_view._main_viewmodel.thread_train.out_folder_train
    with qtbot.waitSignal(window.main_view.dialog_train.on_train_start(out_folder_train, 10),timeout=20000, raising=False) as blocker:
        print('waiting')

        window.main_view.dialog_train._ui.pushbutton_step2_next.click()

    with qtbot.waitSignal(window.main_view.dialog_train.on_validation_start(),timeout=10000, raising=False) as blocker:
        print('waiting')

    redict_result = 'a'
    predict_class ='b'
    filename_list ='c'
    classes ='d'
    out_folder_valid = 'e'
    out_folder_valid_all_set ='f'
    with qtbot.waitSignal(window.main_view.dialog_train.on_validation_complete(redict_result,
                                                                               predict_class,
                                                                                filename_list,
                                                                                classes,
                                                                                out_folder_valid,
                                                                                out_folder_valid_all_set), timeout=10000, raising=False) as blocker:

        print('waiting')
        window.main_view.dialog_train._ui.pushbutton_step3_complete.click()