Skip to content

Commit 24a6b28

Browse files
Scott Sandersonrgbkrk
authored andcommitted
MAINT: Fix namedtuple tests.
1 parent 28070bb commit 24a6b28

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

tests/cloudpickle_test.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -702,13 +702,18 @@ def test_function_module_name(self):
702702
self.assertEqual(pickle_depickle(func).__module__, func.__module__)
703703

704704
def test_namedtuple(self):
705-
MyTuple = collections.namedtuple('MyTuple', ['a', 'b', 'c'])
706705

706+
MyTuple = collections.namedtuple('MyTuple', ['a', 'b', 'c'])
707707
t = MyTuple(1, 2, 3)
708-
depickled_t = pickle_depickle(t)
708+
709+
depickled_t, depickled_MyTuple = pickle_depickle([t, MyTuple])
710+
self.assertIsInstance(depickled_t, depickled_MyTuple)
709711

710712
self.assertEqual((depickled_t.a, depickled_t.b, depickled_t.c), (1, 2, 3))
711-
self.assertEqual(vars(t), vars(depickled_t))
713+
self.assertEqual((depickled_t[0], depickled_t[1], depickled_t[2]), (1, 2, 3))
714+
715+
self.assertEqual(depickled_MyTuple.__name__, 'MyTuple')
716+
self.assertTrue(issubclass(depickled_MyTuple, tuple))
712717

713718
if __name__ == '__main__':
714719
unittest.main()

0 commit comments

Comments
 (0)