diff --git a/tests/cloudpickle_test.py b/tests/cloudpickle_test.py index ee9dff628..2d0f12cd0 100644 --- a/tests/cloudpickle_test.py +++ b/tests/cloudpickle_test.py @@ -653,13 +653,18 @@ def __init__(self, x): self.assertEqual(set(weakset), set([depickled1, depickled2])) def test_namedtuple(self): - MyTuple = collections.namedtuple('MyTuple', ['a', 'b', 'c']) + MyTuple = collections.namedtuple('MyTuple', ['a', 'b', 'c']) t = MyTuple(1, 2, 3) - depickled_t = pickle_depickle(t) + + depickled_t, depickled_MyTuple = pickle_depickle([t, MyTuple]) + self.assertIsInstance(depickled_t, depickled_MyTuple) self.assertEqual((depickled_t.a, depickled_t.b, depickled_t.c), (1, 2, 3)) - self.assertEqual(vars(t), vars(depickled_t)) + self.assertEqual((depickled_t[0], depickled_t[1], depickled_t[2]), (1, 2, 3)) + + self.assertEqual(depickled_MyTuple.__name__, 'MyTuple') + self.assertTrue(issubclass(depickled_MyTuple, tuple)) if __name__ == '__main__':