def test_create_missions_from_json(self): import json f = open("mission_data.json", 'r') json_data = json.load(f) mission_factory = MissionFactory() missions = mission_factory.create_missions_from_file( "mission_data.json") index = 0 for mission_name, mission in missions.items(): self.assertEqual(mission_name, json_data['data'][index]['name']) if isinstance(mission, SlowMission): mission_type = 'slow' elif isinstance(mission, FastMission): mission_type = 'fast' elif isinstance(mission, VerboseFastMission): mission_type = 'verbosefast' self.assertEqual(mission_type, json_data['data'][index]['type']) index += 1
def test_create_missions_from_csv(self): import csv rows = [] with open("mission_data.csv", "r") as csvfile: mission_data_reader = csv.reader(csvfile, delimiter=",") for row in mission_data_reader: rows.append(row) rows.pop(0) mission_factory = MissionFactory() missions = mission_factory.create_missions_from_file( "mission_data.csv") index = 0 for mission_name, mission in missions.items(): self.assertEqual(mission_name, rows[index][0]) if isinstance(mission, SlowMission): mission_type = 'slow' elif isinstance(mission, FastMission): mission_type = 'fast' elif isinstance(mission, VerboseFastMission): mission_type = 'verbosefast' self.assertEqual(mission_type, rows[index][2]) index += 1