Mercurial > repos > shellac > sam_consensus_v3
view env/lib/python3.9/site-packages/prov/tests/utility.py @ 0:4f3585e2f14b draft default tip
"planemo upload commit 60cee0fc7c0cda8592644e1aad72851dec82c959"
author | shellac |
---|---|
date | Mon, 22 Mar 2021 18:12:50 +0000 |
parents | |
children |
line wrap: on
line source
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)