Merge pull request #104 from smikes/monkeyYaml-loadfix

load monkeyYaml without assuming it is on path
This commit is contained in:
Brian Terlson 2014-10-24 14:41:40 -07:00
commit 1f2812adeb
2 changed files with 30 additions and 4 deletions

View File

@ -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()

View File

@ -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):