def setUp(self): dest_array = np.ma.array( [b'FDSL', b'FDSL', b'FDSL', b'FDSL', b'ABCD', b'ABCD'], mask=[True, False, False, False, False, True]) self.dest = P('Destination', array=dest_array) self.afr_dest = A('AFR Destination Airport', value={'id': 2000}) self.node = DestinationAirport()
class TestDestinationAirport(unittest.TestCase): def test_can_operate(self): self.assertEqual(DestinationAirport.get_operational_combinations(), [('Destination',), ('AFR Destination Airport',), ('Destination', 'AFR Destination Airport')]) def setUp(self): dest_array = np.ma.array( [b'FDSL', b'FDSL', b'FDSL', b'FDSL', b'ABCD', b'ABCD'], mask=[True, False, False, False, False, True]) self.dest = P('Destination', array=dest_array) self.afr_dest = A('AFR Destination Airport', value={'id': 2000}) self.node = DestinationAirport() @patch('analysis_engine.api_handler.FileHandler.get_airport') def test_derive_dest(self, get_airport): self.node.derive(self.dest, None) self.assertEqual(self.node.value, get_airport.return_value) get_airport.assert_called_once_with('FDSL') def test_derive_afr_dest(self): self.node.derive(None, self.afr_dest) self.assertEqual(self.node.value, self.afr_dest.value) @patch('analysis_engine.api_handler.FileHandler.get_airport') def test_derive_both(self, get_airport): self.node.derive(self.dest, self.afr_dest) self.assertEqual(self.node.value, get_airport.return_value) get_airport.assert_called_once_with('FDSL') def test_derive_invalid(self): dest_array = np.ma.array( ['000', '0000', '0000', '0000', '00', '0000'], mask=[True, False, False, False, False, True]) dest = P('Destination', array=dest_array) self.node.derive(dest, None) self.assertEqual(self.node.value, None)
def test_can_operate(self): self.assertEqual(DestinationAirport.get_operational_combinations(), [('Destination',), ('AFR Destination Airport',), ('Destination', 'AFR Destination Airport')])