44import unittest
55from enum import Enum , EnumValue
66
7- class TestEnum (unittest .TestCase ):
87
8+ class TestEnum (unittest .TestCase ):
99 def setUp (self ):
1010 # Define standard Enum classes for testing
1111 class Color (Enum ):
1212 RED = 'red'
1313 GREEN = 'green'
1414 BLUE = 'blue'
15-
15+
1616 class Status (Enum ):
1717 IDLE = 0
1818 RUNNING = 1
@@ -42,7 +42,7 @@ def test_immutability(self):
4242 # Test EnumValue immutability
4343 with self .assertRaises (AttributeError ):
4444 self .color_inst .RED .value = 'blue'
45-
45+
4646 # Test Enum instance immutability (once initialized)
4747 with self .assertRaises (AttributeError ):
4848 self .color_inst .NEW_COLOR = 'yellow'
@@ -63,24 +63,25 @@ def test_reverse_lookup_via_instance_call(self):
6363 member = self .color_inst ('green' )
6464 self .assertEqual (member .name , 'GREEN' )
6565
66- with self .assertRaises (ValueError ):
66+ with self .assertRaises (AttributeError ):
6767 self .color_inst ('yellow' )
6868
6969 def test_iteration (self ):
7070 """Test that the Enum instance is iterable and returns all members."""
7171 members = list (self .status_inst )
7272 names = [m .name for m in members ]
7373 values = [m .value for m in members ]
74-
74+
7575 self .assertEqual (len (members ), 3 )
7676 self .assertIn ('IDLE' , names )
7777 self .assertIn (2 , values )
7878
79+
7980# def test_functional_api(self):
8081# """Test dynamic Enum creation using the Functional API."""
8182# # Dynamic creation via Enum(name, names_dict)
8283# State = Enum(name='State', names={'ON': 1, 'OFF': 0})
83- #
84+ #
8485# self.assertTrue(hasattr(State, 'ON'))
8586# self.assertEqual(State.ON.value, 1)
8687# self.assertEqual(State.OFF.name, 'OFF')
@@ -92,18 +93,18 @@ def test_iteration(self):
9293# print('c_repr', c_repr)
9394# c_restored = eval(c_repr)
9495# self.assertEqual(self.color_inst, c_restored)
95- #
96+ #
9697# # Test dynamically created instance
9798# s_dynamic = Enum(name='State', names={'ON': 1, 'OFF': 0})
9899# s_repr = repr(s_dynamic)
99100# s_restored = eval(s_repr)
100101# self.assertEqual(s_dynamic, s_restored)
101102
102- def test_len_and_list_members (self ):
103- """Test utility functions like __len__ and list_members ."""
103+ def test_len_and_list (self ):
104+ """Test utility functions like __len__ and list ."""
104105 self .assertEqual (len (self .color_inst ), 3 )
105- members_list = self .color_inst .list_members ()
106- self .assertEqual (members_list , [( ' BLUE' , 'blue' ), ( ' GREEN' , 'green' ), ( ' RED' , 'red' ) ])
106+ members_list = self .color_inst .list ()
107+ self .assertEqual (members_list , [self . color_inst . BLUE , self . color_inst . GREEN , self . color_inst . RED ])
107108
108109 def test_deletion_prevention (self ):
109110 """Verify that members cannot be deleted."""
0 commit comments