Mercurial > repos > guerler > springsuite
diff planemo/lib/python3.7/site-packages/schema_salad/tests/matcher.py @ 1:56ad4e20f292 draft
"planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"
author | guerler |
---|---|
date | Fri, 31 Jul 2020 00:32:28 -0400 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/planemo/lib/python3.7/site-packages/schema_salad/tests/matcher.py Fri Jul 31 00:32:28 2020 -0400 @@ -0,0 +1,40 @@ +# Copyright (C) The Arvados Authors. All rights reserved. +# +# SPDX-License-Identifier: Apache-2.0 + +import difflib +import re + +from schema_salad.utils import json_dumps + + +class JsonDiffMatcher(object): + """Raise AssertionError with a readable JSON diff when not __eq__(). + + Used with assert_called_with() so it's possible for a human to see + the differences between expected and actual call arguments that + include non-trivial data structures. + """ + + def __init__(self, expected): + self.expected = expected + + def __eq__(self, actual): + expected_json = json_dumps(self.expected, sort_keys=True, indent=2) + actual_json = json_dumps(actual, sort_keys=True, indent=2) + if expected_json != actual_json: + raise AssertionError( + "".join( + difflib.context_diff( + expected_json.splitlines(1), + actual_json.splitlines(1), + fromfile="Expected", + tofile="Actual", + ) + ) + ) + return True + + +def StripYAMLComments(yml): + return re.sub(r"(?ms)^(#.*?\n)*\n*", "", yml)