def test_race_and_no_type(self): race_type = 'Machine' result = DataImport.get_monster_race_and_type(race_type) expected_result: dict = { 'race': 'Machine', 'monster_type': 'Normal Monster' } self.assertEqual(result, expected_result)
def test_race_and_multiple_types(self): race_type = 'Machine / Union / Effect' result = DataImport.get_monster_race_and_type(race_type) expected_result: dict = { 'race': 'Machine', 'monster_type': 'Union Effect Monster' } self.assertEqual(result, expected_result)