def test_turning(self):
self.test_shields = generate_shields(50, 40, 40, 40)
self.test_armors = generate_armor(100, 60, 60, 100)
self.test_weapons = FighterWeapons()
self.test_weapons.add_weapon(
"bow",
Weapon("MDC 8", "MDC",
dict([(1, 8), (2, 8), (3, 8), (4, 8), (5, 8), (6, 8)]), 6,
24, 108000))
self.fighter = Fighter("Avenger", "Heavy Fighter", 175, 3552300, 0,
1200, 1200, 7, True, False, self.test_shields,
self.test_armors, self.test_weapons)
self.assertEqual(0, self.fighter.heading, "Starting heading must be 0")
self.fighter.turn_right()
self.assertEqual(5, self.fighter.heading,
"Heading must have turned right")
self.fighter.turn_right()
self.assertEqual(4, self.fighter.heading,
"Heading must have turned right")
self.fighter.turn_left()
self.assertEqual(5, self.fighter.heading,
"Heading must have turned left")
self.fighter.turn_left()
self.assertEqual(0, self.fighter.heading,
"Heading must have turned left")
self.fighter.turn_left()
self.assertEqual(1, self.fighter.heading,
"Heading must have turned left")
def test_initialization(self):
self.test_shields = generate_shields(50, 40, 40, 40)
self.test_armors = generate_armor(100, 60, 60, 100)
self.test_weapons = FighterWeapons()
self.test_weapons.add_weapon(
"bow",
Weapon("MDC 8", "MDC",
dict([(1, 8), (2, 8), (3, 8), (4, 8), (5, 8), (6, 8)]), 6,
24, 108000))
self.fighter = Fighter("Avenger", "Heavy Fighter", 175, 3552300, 0,
1200, 1200, 7, True, False, self.test_shields,
self.test_armors, self.test_weapons)
self.assertEqual("Avenger", self.fighter.name,
"Fighter class must match")
self.assertEqual("Heavy Fighter", self.fighter.fighter_class,
"Fighter class must match")
self.assertEqual(175, self.fighter.mass, "Mass must match")
self.assertEqual(3552300, self.fighter.cost, "Cost must match")
self.assertEqual(0, self.fighter.center_engine,
"Center engine must match")
self.assertEqual(1200, self.fighter.right_engine,
"Right engine must match")
self.assertEqual(1200, self.fighter.left_engine,
"Left engine must match")
self.assertEqual(7, self.fighter.thrust, "Thrust must match")
self.assertEqual(True, self.fighter.streamlining,
"Streamlining must match")
self.assertEqual(False, self.fighter.antigrav, "Antigrav must match")
self.assertEqual(self.test_shields, self.fighter.shields,
"Shields must match")
self.assertEqual(self.test_armors, self.fighter.armors,
"Armor must match")
self.assertEqual(self.test_weapons, self.fighter.weapons,
"Weapons must match")
def load_fighters(filename, loaded_weapons, image_offset="../"):
fighters_dict = dict()
with open(filename) as json_file:
json_array = json.load(json_file)
for item in json_array:
shields = item['shields']
shield_dict = generate_shields(shields['bow'], shields['left'],
shields['right'], shields['stern'])
armor = item['armor']
armor_dict = generate_armor(armor['bow'], armor['left'],
armor['right'], armor['stern'])
weapons = item['weapons']
fighter_weapons = FighterWeapons()
for weapon in weapons:
if weapon['type'] not in loaded_weapons:
print(
f"ERROR: Weapon Type not found in loaded weapons: {weapon['type']}"
)
fighter_weapons.add_weapon(weapon['location'],
loaded_weapons.get(weapon['type']))
engines = item['engines']
center_engine = engines['center'] if 'center' in engines else 0
left_engine = engines['left'] if 'left' in engines else 0
right_engine = engines['right'] if 'right' in engines else 0
image_filename = os.path.join(THIS_DIR,
image_offset + item['image'])
fighters_dict.update({
item['name']:
Fighter(item['name'], item['class'], item['mass'],
item['cost'], center_engine, right_engine, left_engine,
item['thrust'], item['streamlining'], item['antigrav'],
shield_dict, armor_dict, fighter_weapons,
image_filename)
})
return fighters_dict
def test_add_weapons(self):
self.fighter_weapons = FighterWeapons()
self.bow_weapon = Weapon(
"MDC 8", "MDC",
dict([(1, 8), (2, 8), (3, 8), (4, 8), (5, 8), (6, 8)]), 6, 24,
108000)
self.left_weapon = Weapon(
"NPC 9", "NPC",
dict([(1, 1), (2, 6), (3, 6), (4, 9), (5, 9), (6, 9)]), 7, 16,
104000)
self.right_weapon = Weapon(
"EPC 9", "EPC",
dict([(1, 9), (2, 5), (3, 5), (4, 3), (5, 3), (6, 3)]), 25, 6,
125000)
self.one_weapon = Weapon("1.5/6", "LASER",
dict([(1, 7), (2, 6), (3, 6)]), 10, 10,
120000)
self.two_weapon = Weapon("1.5/3", "LASER",
dict([(1, 4), (2, 3), (3, 3)]), 5, 5, 60000)
self.fighter_weapons.add_weapon("bow", self.bow_weapon)
self.fighter_weapons.add_weapon("left", self.left_weapon)
self.fighter_weapons.add_weapon("right", self.right_weapon)
self.assertEquals(1, self.fighter_weapons.bow.__len__(),
"Bow weapon size must match")
self.assertEquals(1, self.fighter_weapons.left.__len__(),
"Left weapon size must match")
self.assertEquals(1, self.fighter_weapons.right.__len__(),
"Right weapon size must match")
self.assertTrue(self.fighter_weapons.bow.__contains__(self.bow_weapon),
"Bow weapons must include MDC 8")
self.assertTrue(
self.fighter_weapons.left.__contains__(self.left_weapon),
"Left weapons must include NPC 9")
self.assertTrue(
self.fighter_weapons.right.__contains__(self.right_weapon),
"Right weapons must include EPC 9")
def test_moving_forward(self):
self.test_shields = generate_shields(50, 40, 40, 40)
self.test_armors = generate_armor(100, 60, 60, 100)
self.test_weapons = FighterWeapons()
self.test_weapons.add_weapon(
"bow",
Weapon("MDC 8", "MDC",
dict([(1, 8), (2, 8), (3, 8), (4, 8), (5, 8), (6, 8)]), 6,
24, 108000))
self.fighter = Fighter("Avenger", "Heavy Fighter", 175, 3552300, 0,
1200, 1200, 7, True, False, self.test_shields,
self.test_armors, self.test_weapons)
self.assertEqual(0, self.fighter.heading, "Starting heading must be 0")
self.assertEqual(0, self.fighter.x, "Starting x position must be 0")
self.assertEqual(0, self.fighter.y, "Starting y position must be 0")
self.fighter.move_forward()
self.assertEqual(1, self.fighter.x, "New x position must be 1")
self.assertEqual(0, self.fighter.y, "New y position must be 0")
self.fighter.move_forward()
self.assertEqual(2, self.fighter.x, "New x position must be 2")
self.assertEqual(0, self.fighter.y, "New y position must be 0")
self.fighter.turn_right()
self.fighter.move_forward()
self.assertEqual(2, self.fighter.x, "New x position must be 2")
self.assertEqual(1, self.fighter.y, "New y position must be 1")
self.fighter.move_forward()
self.assertEqual(3, self.fighter.x, "New x position must be 3")
self.assertEqual(2, self.fighter.y, "New y position must be 2")
self.fighter.turn_right()
self.fighter.move_forward()
self.assertEqual(2, self.fighter.x, "New x position must be 2")
self.assertEqual(3, self.fighter.y, "New y position must be 3")
self.fighter.move_forward()
self.assertEqual(2, self.fighter.x, "New x position must be 2")
self.assertEqual(4, self.fighter.y, "New y position must be 4")
self.fighter.turn_right()
self.fighter.move_forward()
self.assertEqual(1, self.fighter.x, "New x position must be 1")
self.assertEqual(4, self.fighter.y, "New y position must be 4")
self.fighter.move_forward()
self.assertEqual(0, self.fighter.x, "New x position must be 0")
self.assertEqual(4, self.fighter.y, "New y position must be 4")
self.fighter.turn_right()
self.fighter.move_forward()
self.assertEqual(-1, self.fighter.x, "New x position must be -1")
self.assertEqual(3, self.fighter.y, "New y position must be 3")
self.fighter.move_forward()
self.assertEqual(-1, self.fighter.x, "New x position must be -1")
self.assertEqual(2, self.fighter.y, "New y position must be 2")
self.fighter.turn_right()
self.fighter.move_forward()
self.assertEqual(-1, self.fighter.x, "New x position must be -1")
self.assertEqual(1, self.fighter.y, "New y position must be 1")
self.fighter.move_forward()
self.assertEqual(0, self.fighter.x, "New x position must be 0")
self.assertEqual(0, self.fighter.y, "New y position must be 0")
class FighterTest(unittest.TestCase):
def test_initialization_defaults(self):
self.fighter = Fighter()
self.assertEqual("", self.fighter.name, "Fighter class must match")
self.assertEqual("Light Fighter", self.fighter.fighter_class,
"Fighter class must match")
self.assertEqual(0, self.fighter.mass, "Mass must match")
self.assertEqual(0, self.fighter.cost, "Cost must match")
self.assertEqual(0, self.fighter.center_engine,
"Center engine must match")
self.assertEqual(0, self.fighter.right_engine,
"Right engine must match")
self.assertEqual(0, self.fighter.left_engine, "Left engine must match")
self.assertEqual(0, self.fighter.thrust, "Thrust must match")
self.assertEqual(False, self.fighter.streamlining,
"Streamlining must match")
self.assertEqual(False, self.fighter.antigrav, "Antigrav must match")
self.assertEqual(None, self.fighter.shields, "Shields must match")
self.assertEqual(None, self.fighter.armors, "Armor must match")
self.assertEqual(None, self.fighter.weapons, "Weapons must match")
def test_initialization(self):
self.test_shields = generate_shields(50, 40, 40, 40)
self.test_armors = generate_armor(100, 60, 60, 100)
self.test_weapons = FighterWeapons()
self.test_weapons.add_weapon(
"bow",
Weapon("MDC 8", "MDC",
dict([(1, 8), (2, 8), (3, 8), (4, 8), (5, 8), (6, 8)]), 6,
24, 108000))
self.fighter = Fighter("Avenger", "Heavy Fighter", 175, 3552300, 0,
1200, 1200, 7, True, False, self.test_shields,
self.test_armors, self.test_weapons)
self.assertEqual("Avenger", self.fighter.name,
"Fighter class must match")
self.assertEqual("Heavy Fighter", self.fighter.fighter_class,
"Fighter class must match")
self.assertEqual(175, self.fighter.mass, "Mass must match")
self.assertEqual(3552300, self.fighter.cost, "Cost must match")
self.assertEqual(0, self.fighter.center_engine,
"Center engine must match")
self.assertEqual(1200, self.fighter.right_engine,
"Right engine must match")
self.assertEqual(1200, self.fighter.left_engine,
"Left engine must match")
self.assertEqual(7, self.fighter.thrust, "Thrust must match")
self.assertEqual(True, self.fighter.streamlining,
"Streamlining must match")
self.assertEqual(False, self.fighter.antigrav, "Antigrav must match")
self.assertEqual(self.test_shields, self.fighter.shields,
"Shields must match")
self.assertEqual(self.test_armors, self.fighter.armors,
"Armor must match")
self.assertEqual(self.test_weapons, self.fighter.weapons,
"Weapons must match")
def test_turning(self):
self.test_shields = generate_shields(50, 40, 40, 40)
self.test_armors = generate_armor(100, 60, 60, 100)
self.test_weapons = FighterWeapons()
self.test_weapons.add_weapon(
"bow",
Weapon("MDC 8", "MDC",
dict([(1, 8), (2, 8), (3, 8), (4, 8), (5, 8), (6, 8)]), 6,
24, 108000))
self.fighter = Fighter("Avenger", "Heavy Fighter", 175, 3552300, 0,
1200, 1200, 7, True, False, self.test_shields,
self.test_armors, self.test_weapons)
self.assertEqual(0, self.fighter.heading, "Starting heading must be 0")
self.fighter.turn_right()
self.assertEqual(5, self.fighter.heading,
"Heading must have turned right")
self.fighter.turn_right()
self.assertEqual(4, self.fighter.heading,
"Heading must have turned right")
self.fighter.turn_left()
self.assertEqual(5, self.fighter.heading,
"Heading must have turned left")
self.fighter.turn_left()
self.assertEqual(0, self.fighter.heading,
"Heading must have turned left")
self.fighter.turn_left()
self.assertEqual(1, self.fighter.heading,
"Heading must have turned left")
def test_moving_forward(self):
self.test_shields = generate_shields(50, 40, 40, 40)
self.test_armors = generate_armor(100, 60, 60, 100)
self.test_weapons = FighterWeapons()
self.test_weapons.add_weapon(
"bow",
Weapon("MDC 8", "MDC",
dict([(1, 8), (2, 8), (3, 8), (4, 8), (5, 8), (6, 8)]), 6,
24, 108000))
self.fighter = Fighter("Avenger", "Heavy Fighter", 175, 3552300, 0,
1200, 1200, 7, True, False, self.test_shields,
self.test_armors, self.test_weapons)
self.assertEqual(0, self.fighter.heading, "Starting heading must be 0")
self.assertEqual(0, self.fighter.x, "Starting x position must be 0")
self.assertEqual(0, self.fighter.y, "Starting y position must be 0")
self.fighter.move_forward()
self.assertEqual(1, self.fighter.x, "New x position must be 1")
self.assertEqual(0, self.fighter.y, "New y position must be 0")
self.fighter.move_forward()
self.assertEqual(2, self.fighter.x, "New x position must be 2")
self.assertEqual(0, self.fighter.y, "New y position must be 0")
self.fighter.turn_right()
self.fighter.move_forward()
self.assertEqual(2, self.fighter.x, "New x position must be 2")
self.assertEqual(1, self.fighter.y, "New y position must be 1")
self.fighter.move_forward()
self.assertEqual(3, self.fighter.x, "New x position must be 3")
self.assertEqual(2, self.fighter.y, "New y position must be 2")
self.fighter.turn_right()
self.fighter.move_forward()
self.assertEqual(2, self.fighter.x, "New x position must be 2")
self.assertEqual(3, self.fighter.y, "New y position must be 3")
self.fighter.move_forward()
self.assertEqual(2, self.fighter.x, "New x position must be 2")
self.assertEqual(4, self.fighter.y, "New y position must be 4")
self.fighter.turn_right()
self.fighter.move_forward()
self.assertEqual(1, self.fighter.x, "New x position must be 1")
self.assertEqual(4, self.fighter.y, "New y position must be 4")
self.fighter.move_forward()
self.assertEqual(0, self.fighter.x, "New x position must be 0")
self.assertEqual(4, self.fighter.y, "New y position must be 4")
self.fighter.turn_right()
self.fighter.move_forward()
self.assertEqual(-1, self.fighter.x, "New x position must be -1")
self.assertEqual(3, self.fighter.y, "New y position must be 3")
self.fighter.move_forward()
self.assertEqual(-1, self.fighter.x, "New x position must be -1")
self.assertEqual(2, self.fighter.y, "New y position must be 2")
self.fighter.turn_right()
self.fighter.move_forward()
self.assertEqual(-1, self.fighter.x, "New x position must be -1")
self.assertEqual(1, self.fighter.y, "New y position must be 1")
self.fighter.move_forward()
self.assertEqual(0, self.fighter.x, "New x position must be 0")
self.assertEqual(0, self.fighter.y, "New y position must be 0")