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()
