def test_swap():
"""
Tests swapping the tonnage on the ship and making sure the ship's values
translate accordingly
"""
ship = Spacecraft(200)
armour = Armour("Titanium Steel")
assert armour.tl == 7
assert armour.protection == 2
assert armour.hull_amount == 0.05
assert armour.cost_by_hull_percentage == 0.05
ship.add_armour(armour)
assert ship.get_remaining_cargo() == 190
assert ship.armour_total == 2
assert ship.get_total_cost() == 8.4
ship.set_tonnage(100)
assert ship.get_remaining_cargo() == 95
assert ship.armour_total == 2
assert ship.get_total_cost() == 2.1
ship.remove_armour(armour)
assert ship.get_remaining_cargo() == 100
assert ship.armour_total == 0
assert ship.get_total_cost() == 2
def test_remove():
"""
Tests removing a piece of armour from the ship
"""
ship = Spacecraft(200)
armour = Armour("Titanium Steel")
assert armour.tl == 7
assert armour.protection == 2
assert armour.hull_amount == 0.05
assert armour.cost_by_hull_percentage == 0.05
ship.add_armour(armour)
assert ship.get_remaining_cargo() == 190
assert ship.armour_total == 2
assert ship.get_total_cost() == 8.4
ship.remove_armour(armour)
assert ship.get_remaining_cargo() == 200
assert ship.armour_total == 0
assert ship.get_total_cost() == 8
# testing for removing a non-attached armour piece
ship.remove_armour(armour)
assert ship.get_remaining_cargo() == 200
assert ship.armour_total == 0
assert ship.get_total_cost() == 8
