예제 #1
0
def test_import_user_module_without_py(import_module, training):
    env = TrainingEnvironment(training)
    env.user_script_name = 'nopy'
    env.import_user_module()
    import_module.assert_called_with('nopy')
예제 #2
0
def test_import_user_module(import_module, training):
    env = TrainingEnvironment(training)
    env.import_user_module()
    import_module.assert_called_with('myscript')