diff options
Diffstat (limited to 'test/sanity/validate-modules/utils.py')
-rw-r--r-- | test/sanity/validate-modules/utils.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/test/sanity/validate-modules/utils.py b/test/sanity/validate-modules/utils.py index 32a7fbb06c..dbda4938c7 100644 --- a/test/sanity/validate-modules/utils.py +++ b/test/sanity/validate-modules/utils.py @@ -71,12 +71,20 @@ class CaptureStd(): return self.stdout.getvalue(), self.stderr.getvalue() -def parse_yaml(value, lineno, module, name): +def parse_yaml(value, lineno, module, name, load_all=False): traces = [] errors = [] data = None + + if load_all: + loader = yaml.safe_load_all + else: + loader = yaml.safe_load + try: - data = yaml.safe_load(value) + data = loader(value) + if load_all: + data = list(data) except yaml.MarkedYAMLError as e: e.problem_mark.line += lineno - 1 e.problem_mark.name = '%s.%s' % (module, name) |