class TestExerciseCategory(unittest.TestCase):
'Test the ExerciseCategory class'
# Before each test
def setUp(self):
self.category = ExerciseCategory("Test category")
# After each test
def tearDown(self):
self.category = None
# Test add_exercise_type
def test_add_exercise_type(self):
self.category.add_exercise_type("ex1", "desc1", 0)
self.category.add_exercise_type("ex2", "desc2", 42)
ex1 = self.category.exercise_type_list[0]
ex2 = self.category.exercise_type_list[1]
# test ex1
self.assertTrue(isinstance(ex1, ExerciseType))
self.assertEqual(ex1.name, "ex1")
self.assertEqual(ex1.description, "desc1")
# test ex2
self.assertTrue(isinstance(ex2, ExerciseType))
self.assertEqual(ex2.name, "ex2")
self.assertEqual(ex2.description, "desc2")
def load_exercises_from_xml(xml_root):
category_list = []
exercise_type_dict = {}
exercises_node = xml_root.find("exercises")
for category_node in exercises_node:
category_name = category_node.get("name")
category = ExerciseCategory(category_name)
category_list.append(category)
for exercise_node in category_node.getchildren():
exercise_name = exercise_node.get("name")
exercise_description = exercise_node.get("description")
exercise_rest_str = int(exercise_node.get("rest"))
exercise_type = category.add_exercise_type(exercise_name, exercise_description, exercise_rest_str)
exercise_type_dict[exercise_name] = exercise_type
return (category_list, exercise_type_dict)
def setUp(self):
self.category = ExerciseCategory("Test category")
