def test_select_class_barbarian(self, mock_input):
self.assertEqual("Barbarian", select_class())
def test_select_class_wizard(self, mock_input):
self.assertEqual("Wizard", select_class())
def test_select_class_cst_student(self, mock_input):
self.assertEqual("CST Student", select_class())
def test_select_class_sorcerer(self, mock_input):
self.assertEqual("Sorcerer", select_class())
def test_select_class_warlock(self, mock_input):
self.assertEqual("Warlock", select_class())
def test_select_class_monk(self, mock_input):
self.assertEqual("Monk", select_class())
def test_select_class_paladin(self, mock_input):
self.assertEqual("Paladin", select_class())
def test_select_class_ranger(self, mock_input):
self.assertEqual("Ranger", select_class())
def test_select_class_rogue(self, mock_input):
self.assertEqual("Rogue", select_class())
def test_select_class_druid(self, mock_input):
self.assertEqual("Druid", select_class())
def test_select_class_cleric(self, mock_input):
self.assertEqual("Cleric", select_class())
def test_select_class_fighter(self, mock_input):
self.assertEqual("Fighter", select_class())