def test_removing():
# Tests removing a screen object from the ship
ship = Spacecraft(200)
assert ship.get_remaining_cargo() == 200
assert ship.get_total_cost() == 8.0
damper = Screen("Nuclear Damper")
ship.modify_screen(damper)
assert ship.get_remaining_cargo() == 150
assert ship.get_total_cost() == 58.0
new_damper = Screen("Nuclear Damper")
ship.modify_screen(new_damper)
assert ship.get_remaining_cargo() == 200
assert ship.get_total_cost() == 8.0
def test_adding():
# Tests adding each screen type
damper = Screen("Nuclear Damper")
meson = Screen("Meson Screen")
extra = Screen("Nuclear Damper")
ship = Spacecraft(200)
assert ship.get_remaining_cargo() == 200
assert ship.get_total_cost() == 8.0
ship.modify_screen(damper)
assert ship.get_remaining_cargo() == 150
assert ship.get_total_cost() == 58.0
ship.modify_screen(meson)
assert ship.get_remaining_cargo() == 100
assert ship.get_total_cost() == 118.0