Пример #1
0
 def test_enum(self) -> None:
     lst = list(Color)
     self.assertEqual(len(lst), len(Color))
     self.assertEqual(len(Color), 3)
     self.assertEqual([Color.red, Color.blue, Color.green], lst)
     for i, color in enumerate("red blue green".split(), 0):
         e = Color(i)
         self.assertEqual(e, getattr(Color, color))
         self.assertEqual(e.value, i)
         self.assertNotEqual(e, i)
         self.assertEqual(e.name, color)
         self.assertIn(e, Color)
         self.assertIs(type(e), Color)
         self.assertIsInstance(e, Color)
         self.assertEqual(str(e), "Color." + color)
         self.assertEqual(int(e), i)
         self.assertEqual(repr(e), f"<Color.{color}: {i}>")
Пример #2
0
 def test_enum_in_enum_out(self) -> None:
     self.assertIs(Color(Color.blue), Color.blue)