def test_thing_with_queue(self): obj = ThingWithQueue() flattened = self.pickler.flatten(obj) restored = self.unpickler.restore(flattened) self.assertEqual(type(restored.child_1), type(queue.Queue())) self.assertEqual(type(restored.child_2), type(queue.Queue())) # Check references self.assertTrue(restored.child_1 is restored.childref_1) self.assertTrue(restored.child_2 is restored.childref_2)
def __init__(self): self.child_1 = queue.Queue() self.child_2 = queue.Queue() self.childref_1 = self.child_1 self.childref_2 = self.child_2
def restore(self, data): return queue.Queue()