def test_planet_owner_ship_arrival(self):
user = User('user', Color.RED)
planet = Planet(1, 25, 25, 20, user, None)
units_count = planet.units.count
planet.arrival(user)
self.assertEqual(planet.units.count, units_count + 1)
self.assertEqual(planet.owner, user)
