def test_derive__columbia234(self):
vert_spd = load(
os.path.join(test_data_path,
"gear_on_ground__columbia234_vert_spd.nod"))
torque = load(
os.path.join(test_data_path,
"gear_on_ground__columbia234_torque.nod"))
collective = load(
os.path.join(test_data_path,
"gear_on_ground__columbia234_collective.nod"))
ac_series = A("Series", value="Columbia 234")
wow = GearOnGround()
wow.derive(None, None, vert_spd, torque, ac_series, collective)
self.assertTrue(np.ma.all(wow.array[:252] == 'Ground'))
self.assertTrue(np.ma.all(wow.array[254:540] == 'Air'))
self.assertTrue(np.ma.all(wow.array[1040:1200] == 'Air'))
self.assertTrue(np.ma.all(wow.array[1420:1440] == 'Air'))
self.assertTrue(np.ma.all(wow.array[1533:1550] == 'Ground'))
self.assertTrue(np.ma.all(wow.array[1615:1622] == 'Air'))
#self.assertTrue(np.ma.all(wow.array[1696:1730] == 'Ground'))
self.assertTrue(np.ma.all(wow.array[1900:2150] == 'Air'))
self.assertTrue(np.ma.all(wow.array[2350:2385] == 'Air'))
self.assertTrue(np.ma.all(wow.array[2550:2750] == 'Air'))
self.assertTrue(np.ma.all(wow.array[2900:3020] == 'Air'))
self.assertTrue(np.ma.all(wow.array[3366:3376] == 'Air'))
self.assertTrue(np.ma.all(wow.array[3425:] == 'Ground'))
def test_derive__columbia234_collective(self):
vert_spd = load(os.path.join(test_data_path, "gear_on_ground__columbia234_vert_spd_flight2.nod"))
torque = load(os.path.join(test_data_path, "gear_on_ground__columbia234_torque_flight2.nod"))
collective = load(os.path.join(test_data_path, "gear_on_ground__columbia234_collective_flight2.nod"))
ac_series = A("Series", value="Columbia 234")
wow = GearOnGround()
wow.derive(None, None, vert_spd, torque, ac_series, collective)
self.assertTrue(all(wow.array[:277] == 'Ground'))
self.assertTrue(all(wow.array[300:1272] == 'Air'))
self.assertTrue(all(wow.array[1275:1470] == 'Ground'))
self.assertTrue(all(wow.array[1474:1772] == 'Air'))
self.assertTrue(all(wow.array[1775:1803] == 'Ground'))
self.assertTrue(all(wow.array[1806:2107] == 'Air'))
self.assertTrue(all(wow.array[2109:2200] == 'Ground'))
self.assertTrue(all(wow.array[2203:3894] == 'Air'))
self.assertTrue(all(wow.array[3896:] == 'Ground'))
def test_derive__columbia234_collective(self):
vert_spd = load(os.path.join(test_data_path, "gear_on_ground__columbia234_vert_spd_flight2.nod"))
torque = load(os.path.join(test_data_path, "gear_on_ground__columbia234_torque_flight2.nod"))
collective = load(os.path.join(test_data_path, "gear_on_ground__columbia234_collective_flight2.nod"))
ac_series = A("Series", value="Columbia 234")
wow = GearOnGround()
wow.derive(None, None, vert_spd, torque, ac_series, collective)
self.assertTrue(all(wow.array[:277] == 'Ground'))
self.assertTrue(all(wow.array[300:1272] == 'Air'))
self.assertTrue(all(wow.array[1275:1470] == 'Ground'))
self.assertTrue(all(wow.array[1474:1772] == 'Air'))
self.assertTrue(all(wow.array[1775:1803] == 'Ground'))
self.assertTrue(all(wow.array[1806:2107] == 'Air'))
self.assertTrue(all(wow.array[2109:2200] == 'Ground'))
self.assertTrue(all(wow.array[2203:3894] == 'Air'))
self.assertTrue(all(wow.array[3896:] == 'Ground'))
def test_derive__columbia234(self):
vert_spd = load(os.path.join(test_data_path, "gear_on_ground__columbia234_vert_spd.nod"))
torque = load(os.path.join(test_data_path, "gear_on_ground__columbia234_torque.nod"))
collective = load(os.path.join(test_data_path,"gear_on_ground__columbia234_collective.nod"))
ac_series = A("Series", value="Columbia 234")
wow = GearOnGround()
wow.derive(None, None, vert_spd, torque, ac_series, collective)
self.assertTrue(np.ma.all(wow.array[:252] == 'Ground'))
self.assertTrue(np.ma.all(wow.array[254:540] == 'Air'))
self.assertTrue(np.ma.all(wow.array[1040:1200] == 'Air'))
self.assertTrue(np.ma.all(wow.array[1420:1440] == 'Air'))
self.assertTrue(np.ma.all(wow.array[1533:1550] == 'Ground'))
self.assertTrue(np.ma.all(wow.array[1615:1622] == 'Air'))
#self.assertTrue(np.ma.all(wow.array[1696:1730] == 'Ground'))
self.assertTrue(np.ma.all(wow.array[1900:2150] == 'Air'))
self.assertTrue(np.ma.all(wow.array[2350:2385] == 'Air'))
self.assertTrue(np.ma.all(wow.array[2550:2750] == 'Air'))
self.assertTrue(np.ma.all(wow.array[2900:3020] == 'Air'))
self.assertTrue(np.ma.all(wow.array[3366:3376] == 'Air'))
self.assertTrue(np.ma.all(wow.array[3425:] == 'Ground'))
def open_node_container(zip_path):
'''
Opens a zip file containing nodes.
TODO: Do not compress to the current directory.
'''
with zipfile.ZipFile(zip_path, 'r') as zip_file:
filenames = zip_file.namelist()
zip_file.extractall('.')
nodes = defaultdict(dict)
for filename in filenames:
match = re.match('^(?P<flight_pk>\d+) - (?P<node_name>[\w\d\s]+).nod$', filename)
if not match:
print "Skipping invalid filename '%s'"
os.remove(filename)
continue
groupdict = match.groupdict()
nodes[groupdict['flight_pk']][groupdict['node_name']] = load(filename)
os.remove(filename)
return nodes
def test_derive(self):
flight_number = load(os.path.join(test_data_path,
'FDRFlightNumber_FlightNumber.nod'))
node = FlightNumber()
node.derive(flight_number)
self.assertEqual(node.value, '805')