def setUp(self): self.node_class = TakeoffDatetime self.takeoff_datetime = self.node_class() self.takeoff_datetime.set_flight_attr = Mock() self.start_dt = A('Start Datetime', value=datetime(1970, 1, 1)) self.liftoff = KTI(name='Liftoff', frequency=2, items=[ KeyTimeInstance(index=64), ]) self.rto = buildsection('Rejected Takeoff', 15, 20) self.off_blocks = KTI(name='Off Blocks', frequency=0.25, items=[ KeyTimeInstance(index=3), ])
def test_derive(self): ''' Tests every flow, but does not test every conceivable set of arguments. ''' type_node = FlightType() type_node.set_flight_attr = Mock() # Liftoff and Touchdown. fast = S('Fast', items=[slice(5,10)]) liftoffs = KTI('Liftoff', items=[KeyTimeInstance(5, 'a')]) touchdowns = KTI('Touchdown', items=[KeyTimeInstance(10, 'x')]) type_node.derive(None, fast, None, liftoffs, touchdowns, None, None, None) type_node.set_flight_attr.assert_called_once_with( FlightType.Type.COMPLETE) # Would be 'COMPLETE', but 'AFR Type' overrides it. afr_type = A('AFR Type', value=FlightType.Type.FERRY) type_node.set_flight_attr = Mock() type_node.derive(afr_type, fast, None, liftoffs, touchdowns, None, None, None) type_node.set_flight_attr.assert_called_once_with(FlightType.Type.FERRY) # Liftoff missing. empty_liftoffs = KTI('Liftoff') type_node.set_flight_attr = Mock() try: type_node.derive(None, fast, None, empty_liftoffs, touchdowns, None, None, None) except InvalidFlightType as err: self.assertEqual(err.flight_type, 'TOUCHDOWN_ONLY') # Touchdown missing. empty_touchdowns = KTI('Touchdown') type_node.set_flight_attr = Mock() try: type_node.derive(None, fast, None, liftoffs, empty_touchdowns, None, None, None) except InvalidFlightType as err: self.assertEqual(err.flight_type, 'LIFTOFF_ONLY') # Liftoff and Touchdown missing, only Fast. type_node.set_flight_attr = Mock() type_node.derive(None, fast, None, empty_liftoffs, empty_touchdowns, None, None, None) type_node.set_flight_attr.assert_called_once_with( FlightType.Type.INCOMPLETE) # Liftoff, Touchdown and Fast missing. empty_fast = fast = S('Fast') type_node.set_flight_attr = Mock() type_node.derive(None, empty_fast, None, empty_liftoffs, empty_touchdowns, None, None, None) type_node.set_flight_attr.assert_called_once_with( FlightType.Type.INCOMPLETE) # Liftoff, Touchdown and Fast missing, RTO. rto = buildsection('Rejected Takeoff', 5, 10) type_node.set_flight_attr = Mock() type_node.derive(None, empty_fast, None, empty_liftoffs, empty_touchdowns, None, rto, None) type_node.set_flight_attr.assert_called_once_with( FlightType.Type.REJECTED_TAKEOFF) # Liftoff after Touchdown. late_liftoffs = KTI('Liftoff', items=[KeyTimeInstance(20, 'a')]) type_node.set_flight_attr = Mock() try: type_node.derive(None, fast, None, late_liftoffs, touchdowns, None, None, None) except InvalidFlightType as err: self.assertEqual(err.flight_type, 'TOUCHDOWN_BEFORE_LIFTOFF') # Touch and Go before Touchdown. afr_type = A('AFR Type', value=FlightType.Type.TRAINING) touch_and_gos = KTI('Touch And Go', items=[KeyTimeInstance(7, 'a')]) type_node.set_flight_attr = Mock() type_node.derive(afr_type, fast, None, liftoffs, touchdowns, touch_and_gos, None, None) type_node.set_flight_attr.assert_called_once_with( FlightType.Type.TRAINING) # Touch and Go after Touchdown. afr_type = A('AFR Type', value=FlightType.Type.TRAINING) touch_and_gos = KTI('Touch And Go', items=[KeyTimeInstance(15, 'a')]) type_node.set_flight_attr = Mock() try: type_node.derive(afr_type, fast, None, liftoffs, touchdowns, touch_and_gos, None, None) except InvalidFlightType as err: self.assertEqual(err.flight_type, 'LIFTOFF_ONLY')