summaryrefslogtreecommitdiff
path: root/test/sanity/validate-modules/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/sanity/validate-modules/utils.py')
-rw-r--r--test/sanity/validate-modules/utils.py12
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)