l4/pkg/python/contrib/Lib/test/test_typechecks.py

   1 """Unit tests for __instancecheck__ and __subclasscheck__."""
   2
   3 import unittest
   4 from test import test_support
   5
   6
   7 class ABC(type):
   8
   9     def __instancecheck__(cls, inst):
  10         """Implement isinstance(inst, cls)."""
  11         return any(cls.__subclasscheck__(c)
  12                    for c in set([type(inst), inst.__class__]))
  13
  14     def __subclasscheck__(cls, sub):
  15         """Implement issubclass(sub, cls)."""
  16         candidates = cls.__dict__.get("__subclass__", set()) | set([cls])
  17         return any(c in candidates for c in sub.mro())
  18
  19
  20 class Integer:
  21
  22     __metaclass__ = ABC
  23
  24     __subclass__ = set([int])
  25
  26
  27 class SubInt(Integer):
  28
  29     pass
  30
  31
  32 class Evil:
  33     def __instancecheck__(self, inst): return False
  34
  35
  36 class TypeChecksTest(unittest.TestCase):
  37
  38     def testIsSubclassInternal(self):
  39         self.assertEqual(Integer.__subclasscheck__(int), True)
  40         self.assertEqual(Integer.__subclasscheck__(float), False)
  41
  42     def testIsSubclassBuiltin(self):
  43         self.assertEqual(issubclass(int, Integer), True)
  44         self.assertEqual(issubclass(int, (Integer,)), True)
  45         self.assertEqual(issubclass(float, Integer), False)
  46         self.assertEqual(issubclass(float, (Integer,)), False)
  47
  48     def testIsInstanceBuiltin(self):
  49         self.assertEqual(isinstance(42, Integer), True)
  50         self.assertEqual(isinstance(42, (Integer,)), True)
  51         self.assertEqual(isinstance(3.14, Integer), False)
  52         self.assertEqual(isinstance(3.14, (Integer,)), False)
  53
  54     def testIsInstanceActual(self):
  55         self.assertEqual(isinstance(Integer(), Integer), True)
  56         self.assertEqual(isinstance(Integer(), (Integer,)), True)
  57
  58     def testIsSubclassActual(self):
  59         self.assertEqual(issubclass(Integer, Integer), True)
  60         self.assertEqual(issubclass(Integer, (Integer,)), True)
  61
  62     def testSubclassBehavior(self):
  63         self.assertEqual(issubclass(SubInt, Integer), True)
  64         self.assertEqual(issubclass(SubInt, (Integer,)), True)
  65         self.assertEqual(issubclass(SubInt, SubInt), True)
  66         self.assertEqual(issubclass(SubInt, (SubInt,)), True)
  67         self.assertEqual(issubclass(Integer, SubInt), False)
  68         self.assertEqual(issubclass(Integer, (SubInt,)), False)
  69         self.assertEqual(issubclass(int, SubInt), False)
  70         self.assertEqual(issubclass(int, (SubInt,)), False)
  71         self.assertEqual(isinstance(SubInt(), Integer), True)
  72         self.assertEqual(isinstance(SubInt(), (Integer,)), True)
  73         self.assertEqual(isinstance(SubInt(), SubInt), True)
  74         self.assertEqual(isinstance(SubInt(), (SubInt,)), True)
  75         self.assertEqual(isinstance(42, SubInt), False)
  76         self.assertEqual(isinstance(42, (SubInt,)), False)
  77
  78     def testInfiniteRecursionCaughtProperly(self):
  79         e = Evil()
  80         # This invokes isinstance() recursively, until the stack is exhausted.
  81         self.assertRaises(RuntimeError, isinstance, e, Evil)
  82         # XXX How to check the same situation for issubclass()?
  83
  84
  85 def test_main():
  86     test_support.run_unittest(TypeChecksTest)
  87
  88
  89 if __name__ == "__main__":
  90     unittest.main()