From 9e154cea9769da7ee799ba7c6da47b01254fad78 Mon Sep 17 00:00:00 2001 From: debugtalk Date: Thu, 20 Jul 2017 10:32:58 +0800 Subject: [PATCH] make custom module importable anywhere --- ate/context.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/ate/context.py b/ate/context.py index ed59e790..c3685f90 100644 --- a/ate/context.py +++ b/ate/context.py @@ -1,6 +1,8 @@ import copy import importlib +import os import re +import sys import types from collections import OrderedDict @@ -67,6 +69,7 @@ class Context(object): def import_module_functions(self, modules, level="testcase"): """ import modules and bind all functions within the context """ + sys.path.insert(0, os.getcwd()) for module_name in modules: imported = importlib.import_module(module_name) imported_functions_dict = dict(filter(is_function, vars(imported).items()))