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')
def test_import_user_module(import_module, training): env = TrainingEnvironment(training) env.import_user_module() import_module.assert_called_with('myscript')