def test_specific_plane_to_takeoff(self): jfk = Airport() jfk.dock("Plane A") jfk.dock("Plane B") jfk.takeoff("Plane A") self.assertEqual(jfk.docked[0], "Plane B")
def test_takeoff_removes_plane_from_docked(self): jfk = Airport() jfk.dock("Plane") jfk.takeoff() self.assertEqual(len(jfk.docked), 0)