Simplify tests

Author: sergey-miryanov

Lib/test/test_capi/test_tuple.py

@@ -121,59 +121,34 @@ def test_tuple_pack(self):
         # CRASHES pack(2, [1])
 
     def test_tuple_from_pair(self):
-        # Test _PyTuple_FromPair()
-        tuple_from_pair = _testinternalcapi._tuple_from_pair
-
-        self.assertEqual(tuple_from_pair(1, 2), (1, 2))
-        self.assertEqual(tuple_from_pair(None, None), (None, None))
-        self.assertEqual(tuple_from_pair(True, False), (True, False))
-
-        # user class supports gc
-        class Temp:
-            pass
-        temp = Temp()
-        temp_rc = getrefcount(temp)
-        self.assertEqual(tuple_from_pair(temp, temp), (temp, temp))
-        self.assertEqual(getrefcount(temp), temp_rc)
-
-        self.assertRaises(TypeError, tuple_from_pair, 1, 2, 3)
-        self.assertRaises(TypeError, tuple_from_pair, 1)
-        self.assertRaises(TypeError, tuple_from_pair)
-
-        self.assertFalse(gc.is_tracked(tuple_from_pair(1, 2)))
-        self.assertFalse(gc.is_tracked(tuple_from_pair(None, None)))
-        self.assertFalse(gc.is_tracked(tuple_from_pair(True, False)))
-        self.assertTrue(gc.is_tracked(tuple_from_pair(temp, (1, 2))))
-        self.assertTrue(gc.is_tracked(tuple_from_pair(temp, 1)))
-        self.assertTrue(gc.is_tracked(tuple_from_pair([], {})))
-
-    def test_tuple_from_pair_steal(self):
-        # Test _PyTuple_FromPairSteal()
-        tuple_from_pair = _testinternalcapi._tuple_from_pair_steal
-
-        self.assertEqual(tuple_from_pair(1, 2), (1, 2))
-        self.assertEqual(tuple_from_pair(None, None), (None, None))
-        self.assertEqual(tuple_from_pair(True, False), (True, False))
-
-        # user class supports gc
-        class Temp:
-            pass
-        temp = Temp()
-        temp_rc = getrefcount(temp)
-        self.assertEqual(tuple_from_pair(temp, temp), (temp, temp))
-        self.assertEqual(getrefcount(temp), temp_rc)
-
-        self.assertRaises(TypeError, tuple_from_pair, 1, 2, 3)
-        self.assertRaises(TypeError, tuple_from_pair, 1)
-        self.assertRaises(TypeError, tuple_from_pair)
-
-        self.assertFalse(gc.is_tracked(tuple_from_pair(1, 2)))
-        self.assertFalse(gc.is_tracked(tuple_from_pair(None, None)))
-        self.assertFalse(gc.is_tracked(tuple_from_pair(True, False)))
-        self.assertTrue(gc.is_tracked(tuple_from_pair(temp, (1, 2))))
-        self.assertTrue(gc.is_tracked(tuple_from_pair(temp, 1)))
-        self.assertTrue(gc.is_tracked(tuple_from_pair([], {})))
-
+        # Test _PyTuple_FromPair, _PyTuple_FromPairSteal
+        ctors = (("_PyTuple_FromPair", _testinternalcapi._tuple_from_pair),
+                 ("_PyTuple_FromPairSteal", _testinternalcapi._tuple_from_pair_steal))
+
+        for name, ctor in ctors:
+            with self.subTest(name):
+                self.assertEqual(ctor(1, 2), (1, 2))
+                self.assertEqual(ctor(None, None), (None, None))
+                self.assertEqual(ctor(True, False), (True, False))
+
+                # user class supports gc
+                class Temp:
+                    pass
+                temp = Temp()
+                temp_rc = getrefcount(temp)
+                self.assertEqual(ctor(temp, temp), (temp, temp))
+                self.assertEqual(getrefcount(temp), temp_rc)
+
+                self.assertRaises(TypeError, ctor, 1, 2, 3)
+                self.assertRaises(TypeError, ctor, 1)
+                self.assertRaises(TypeError, ctor)
+
+                self.assertFalse(gc.is_tracked(ctor(1, 2)))
+                self.assertFalse(gc.is_tracked(ctor(None, None)))
+                self.assertFalse(gc.is_tracked(ctor(True, False)))
+                self.assertTrue(gc.is_tracked(ctor(temp, (1, 2))))
+                self.assertTrue(gc.is_tracked(ctor(temp, 1)))
+                self.assertTrue(gc.is_tracked(ctor([], {})))
 
     def test_tuple_size(self):
         # Test PyTuple_Size()