MAINT: Fix namedtuple tests.

cloudpipe · Aug 30, 2017 · 7ccde35 · 7ccde35
1 parent e0d68ac
commit 7ccde35
Showing 1 changed file with 8 additions and 3 deletions.
diff --git 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__':