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)