From 9dc29da89701c2b27eb26268ec9a096d0297f1b6 Mon Sep 17 00:00:00 2001 From: Sam Mikes Date: Fri, 24 Oct 2014 17:44:18 +0100 Subject: [PATCH] load monkeyYaml without assuming it is on path use imp to import monkeyYaml locally make monkeyYaml the backup again --- tools/packaging/parseTestRecord.py | 18 ++++++++++++++++-- tools/packaging/test/test_monkeyYaml.py | 16 ++++++++++++++-- 2 files changed, 30 insertions(+), 4 deletions(-) diff --git a/tools/packaging/parseTestRecord.py b/tools/packaging/parseTestRecord.py index 60590af005..ce4550a0a4 100644 --- a/tools/packaging/parseTestRecord.py +++ b/tools/packaging/parseTestRecord.py @@ -16,6 +16,7 @@ import subprocess import sys import tempfile import time +import imp # from TestCasePackagerConfig import * @@ -115,6 +116,19 @@ def importYamlLoad(): try: import yaml yamlLoad = yaml.load - except ImportError: - import monkeyYaml + except: + monkeyYaml = loadMonkeyYaml() yamlLoad = monkeyYaml.load + +def loadMonkeyYaml(): + f = None + try: + p = os.path.dirname(os.path.realpath(__file__)) + (f, pathname, description) = imp.find_module("monkeyYaml", [p]) + module = imp.load_module("monkeyYaml", f, pathname, description) + return module + except: + raise ImportError("Cannot load monkeyYaml") + finally: + if f: + f.close() diff --git a/tools/packaging/test/test_monkeyYaml.py b/tools/packaging/test/test_monkeyYaml.py index edf7b939e9..ada802356f 100644 --- a/tools/packaging/test/test_monkeyYaml.py +++ b/tools/packaging/test/test_monkeyYaml.py @@ -7,12 +7,24 @@ import unittest import os import yaml +import imp # add parent dir to search path import sys -sys.path.insert(0, "..") +#sys.path.insert(0, "..") -import monkeyYaml +f = None +try: + (f, pathname, description) = imp.find_module("monkeyYaml", [os.path.join(os.getcwd(), "../")]) + module = imp.load_module("monkeyYaml", f, pathname, description) + monkeyYaml = module +except: + raise ImportError("Cannot load monkeyYaml") +finally: + if f: + f.close() + +#import monkeyYaml class TestMonkeyYAMLParsing(unittest.TestCase):