diff env/lib/python3.9/site-packages/prov/tests/utility.py @ 0:4f3585e2f14b draft default tip

"planemo upload commit 60cee0fc7c0cda8592644e1aad72851dec82c959"

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/env/lib/python3.9/site-packages/prov/tests/utility.py	Mon Mar 22 18:12:50 2021 +0000
@@ -0,0 +1,46 @@
+from __future__ import (absolute_import, division, print_function,
+                        unicode_literals)
+
+import io
+import logging
+import unittest
+
+from prov.model import ProvDocument
+
+
+logger = logging.getLogger(__name__)
+
+
+class DocumentBaseTestCase(unittest.TestCase):
+    def do_tests(self, prov_doc, msg=None):
+        pass
+
+
+class RoundTripTestCase(DocumentBaseTestCase):
+    """A serializer test should subclass this class and set the class property
+    FORMAT to the correct value (e.g. 'json', 'xml', 'rdf').
+    """
+    FORMAT = None  # a subclass should change this
+
+    def do_tests(self, prov_doc, msg=None):
+        self.assertRoundTripEquivalence(prov_doc, msg)
+
+    def assertRoundTripEquivalence(self, prov_doc, msg=None):
+        if self.FORMAT is None:
+            # This is a dummy test, just return
+            return
+
+        with io.BytesIO() as stream:
+            prov_doc.serialize(destination=stream, format=self.FORMAT, indent=4)
+            stream.seek(0, 0)
+
+            prov_doc_new = ProvDocument.deserialize(source=stream,
+                                                    format=self.FORMAT)
+            stream.seek(0, 0)
+            # Assume UTF-8 encoding which is forced by the particular
+            # PROV XML implementation and should also work for the PROV
+            # JSON implementation.
+            msg_extra = u"'%s' serialization content:\n%s" % (
+                self.FORMAT, stream.read().decode("utf-8"))
+            msg = u'\n'.join((msg, msg_extra)) if msg else msg_extra
+            self.assertEqual(prov_doc, prov_doc_new, msg)