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')])
