import unittest from shared.defines import Stage, State # class TestCase(unittest.TestCase): # def test_definition_Stage(self): # for stage in Stage: # self.assertEqual(stage.name.lower().capitalize(), stage.value) # def test_definition_State(self): # for state in State: # self.assertEqual(state.name.lower().capitalize(), state.value) if __name__ == '__main__': unittest.main()