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")