Mercurial > repos > shellac > guppy_basecaller
diff env/lib/python3.7/site-packages/prov/tests/utility.py @ 2:6af9afd405e9 draft
"planemo upload commit 0a63dd5f4d38a1f6944587f52a8cd79874177fc1"
author | shellac |
---|---|
date | Thu, 14 May 2020 14:56:58 -0400 |
parents | 26e78fe6e8c4 |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/env/lib/python3.7/site-packages/prov/tests/utility.py Thu May 14 14:56:58 2020 -0400 @@ -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)