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}>")
def test_enum_in_enum_out(self) -> None: self.assertIs(Color(Color.blue), Color.blue)