def test_orbit_zone(): planet = Planet(SUN) assert planet.orbit_zone() == 1 planet.axis = 4 assert planet.orbit_zone() == 2 planet.axis = 22 assert planet.orbit_zone() == 3