Skip to content

Commit

Permalink
Added tests for new methods.
Browse files Browse the repository at this point in the history
  • Loading branch information
lumip committed Jul 12, 2018
1 parent bbc41b4 commit c8a106d
Showing 1 changed file with 29 additions and 3 deletions.
32 changes: 29 additions & 3 deletions tests/serialization_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@

from qctoolkit.serialization import FilesystemBackend, CachingBackend, Serializable, JSONSerializableEncoder,\
ZipFileBackend, AnonymousSerializable, DictBackend, PulseStorage, JSONSerializableDecoder, Serializer,\
get_default_pulse_registry, set_default_pulse_registry, SerializableMeta
get_default_pulse_registry, set_default_pulse_registry, new_default_pulse_registry, SerializableMeta

from qctoolkit.expressions import ExpressionScalar

Expand Down Expand Up @@ -483,6 +483,33 @@ def get_type_identifier(cls):
self.assertEqual(SerializableMeta.deserialization_callbacks['foo.bar.never'], NativeDeserializable)


class DefaultPulseRegistryManipulationTests(unittest.TestCase):

def test_get_set_default_pulse_registry(self) -> None:
# store previous registry
previous_registry = get_default_pulse_registry()

registry = dict()
set_default_pulse_registry(registry)
self.assertIs(get_default_pulse_registry(), registry)

# restore previous registry
set_default_pulse_registry(previous_registry)
self.assertIs(get_default_pulse_registry(), previous_registry)

def test_new_default_pulse_registry(self) -> None:
# store previous registry
previous_registry = get_default_pulse_registry()

new_default_pulse_registry()
self.assertIsNotNone(get_default_pulse_registry())
self.assertIsNot(get_default_pulse_registry(), previous_registry)

# restore previous registry
set_default_pulse_registry(previous_registry)
self.assertIs(get_default_pulse_registry(), previous_registry)


class PulseStorageTests(unittest.TestCase):
def setUp(self):
self.backend = DummyStorageBackend()
Expand Down Expand Up @@ -633,8 +660,7 @@ def test_set_to_default_registry(self) -> None:
pulse_storage.set_to_default_registry()
self.assertIs(get_default_pulse_registry(), pulse_storage)
finally:
import qctoolkit.serialization
qctoolkit.serialization.default_pulse_registry = previous_default_registry
set_default_pulse_registry(previous_default_registry)

def test_beautified_json(self) -> None:
data = {'e': 89, 'b': 151, 'c': 123515, 'a': 123, 'h': 2415}
Expand Down

0 comments on commit c8a106d

Please sign in to comment.