kopia lustrzana https://github.com/glidernet/ogn-python
47 wiersze
2.0 KiB
Python
47 wiersze
2.0 KiB
Python
import datetime
|
|
import unittest
|
|
|
|
from tests.base import TestBaseDB, db
|
|
|
|
from app.model import TakeoffLanding
|
|
|
|
from app.collect.takeoff_landings import update_entries
|
|
|
|
|
|
class TestTakeoffLanding(TestBaseDB):
|
|
def test_broken_rope(self):
|
|
"""The algorithm should detect one takeoff and one landing."""
|
|
|
|
self.insert_airports_and_devices()
|
|
self.insert_aircraft_beacons_broken_rope()
|
|
|
|
# find the takeoff and the landing
|
|
update_entries(db.session, start=datetime.datetime(2016, 7, 2, 0, 0, 0), end=datetime.datetime(2016, 7, 2, 23, 59, 59))
|
|
takeoff_landing_query = db.session.query(TakeoffLanding).filter(db.between(TakeoffLanding.timestamp, datetime.datetime(2016, 7, 2, 0, 0, 0), datetime.datetime(2016, 7, 2, 23, 59, 59)))
|
|
|
|
self.assertEqual(len(takeoff_landing_query.all()), 2)
|
|
for entry in takeoff_landing_query.all():
|
|
self.assertEqual(entry.airport.name, "Koenigsdorf")
|
|
|
|
# we should not find the takeoff and the landing again
|
|
update_entries(db.session, start=datetime.datetime(2016, 7, 2, 0, 0, 0), end=datetime.datetime(2016, 7, 2, 23, 59, 59))
|
|
self.assertEqual(len(takeoff_landing_query.all()), 2)
|
|
|
|
def test_broken_rope_with_stall(self):
|
|
"""Here we have a broken rope where the glider passes again the threshold for take off."""
|
|
|
|
self.insert_airports_and_devices()
|
|
self.insert_aircraft_beacons_broken_rope_with_stall()
|
|
|
|
# find the takeoff and the landing
|
|
update_entries(db.session, start=datetime.datetime(2019, 4, 13, 0, 0, 0), end=datetime.datetime(2019, 4, 13, 23, 59, 59))
|
|
takeoff_landings = db.session.query(TakeoffLanding).filter(db.between(TakeoffLanding.timestamp, datetime.datetime(2019, 4, 13, 0, 0, 0), datetime.datetime(2019, 4, 13, 23, 59, 59))).all()
|
|
|
|
self.assertEqual(len(takeoff_landings), 2)
|
|
for entry in takeoff_landings:
|
|
self.assertEqual(entry.airport.name, "Koenigsdorf")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|