def setUp(self):
self.user = User.objects.create_user('testuser', '*****@*****.**',
'testpass')
self.user.save()
# Mission
pos = GpsPosition()
pos.latitude = 10
pos.longitude = 100
pos.save()
apos = AerialPosition()
apos.latitude = 10
apos.longitude = 100
apos.altitude_msl = 1000
apos.save()
wpt = Waypoint()
wpt.latitude = 10
wpt.longitude = 100
wpt.altitude_msl = 1000
wpt.order = 10
wpt.save()
self.mission = MissionConfig()
self.mission.home_pos = pos
self.mission.emergent_last_known_pos = pos
self.mission.off_axis_odlc_pos = pos
self.mission.air_drop_pos = pos
self.mission.save()
self.mission.mission_waypoints.add(wpt)
self.mission.search_grid_points.add(wpt)
self.mission.save()
self.now = timezone.now()
def setUp(self):
"""Sets up the tests."""
super(TestOdlc, self).setUp()
self.user = User.objects.create_user('user', '*****@*****.**',
'pass')
# Mission
pos = GpsPosition()
pos.latitude = 10
pos.longitude = 100
pos.save()
wpt = Waypoint()
wpt.latitude = 10
wpt.longitude = 100
wpt.altitude_msl = 1000
wpt.order = 10
wpt.save()
self.mission = MissionConfig()
self.mission.home_pos = pos
self.mission.lost_comms_pos = pos
self.mission.emergent_last_known_pos = pos
self.mission.off_axis_odlc_pos = pos
self.mission.map_center_pos = pos
self.mission.map_height_ft = 1
self.mission.air_drop_pos = pos
self.mission.ugv_drive_pos = pos
self.mission.save()
self.mission.mission_waypoints.add(wpt)
self.mission.search_grid_points.add(wpt)
self.mission.save()
def create_data(self):
"""Create a basic sample dataset."""
self.user1 = User.objects.create_user('user1', '*****@*****.**',
'testpass')
self.user1.save()
self.user2 = User.objects.create_user('user2', '*****@*****.**',
'testpass')
self.user2.save()
# Mission
pos = GpsPosition()
pos.latitude = 10
pos.longitude = 100
pos.save()
wpt = Waypoint()
wpt.order = 10
wpt.latitude = 10
wpt.longitude = 100
wpt.altitude_msl = 1000
wpt.save()
self.mission = MissionConfig()
self.mission.home_pos = pos
self.mission.emergent_last_known_pos = pos
self.mission.off_axis_odlc_pos = pos
self.mission.air_drop_pos = pos
self.mission.save()
self.mission.mission_waypoints.add(wpt)
self.mission.search_grid_points.add(wpt)
self.mission.save()
# user1 is flying
event = TakeoffOrLandingEvent(user=self.user1,
mission=self.mission,
uas_in_air=True)
event.save()
# user2 has landed
event = TakeoffOrLandingEvent(user=self.user2,
mission=self.mission,
uas_in_air=True)
event.save()
event = TakeoffOrLandingEvent(user=self.user2,
mission=self.mission,
uas_in_air=False)
event.save()
# user2 is active
self.timestamp = timezone.now()
self.telem = UasTelemetry(user=self.user2,
latitude=38.6462,
longitude=-76.2452,
altitude_msl=0,
uas_heading=90)
self.telem.save()
self.telem.timestamp = dateutil.parser.parse(
u'2016-10-01T00:00:00.0+00:00')
self.telem.save()
def setUp(self):
"""Creates test data."""
# Form test set for contains position
self.testdata_containspos = []
for test_data in TESTDATA_FLYZONE_CONTAINSPOS:
# Create the FlyZone
zone = FlyZone()
zone.altitude_msl_min = test_data['min_alt']
zone.altitude_msl_max = test_data['max_alt']
zone.save()
for waypoint_id in range(len(test_data['waypoints'])):
(lat, lon) = test_data['waypoints'][waypoint_id]
wpt = Waypoint()
wpt.order = waypoint_id
wpt.latitude = lat
wpt.longitude = lon
wpt.altitude_msl = 0
wpt.save()
zone.boundary_pts.add(wpt)
# Form test set
test_pos = []
for pos in test_data['inside_pos']:
test_pos.append((pos, True))
for pos in test_data['outside_pos']:
test_pos.append((pos, False))
# Store
self.testdata_containspos.append((zone, test_pos))
def setUp(self):
# Mission
pos = GpsPosition()
pos.latitude = 10
pos.longitude = 100
pos.save()
wpt = Waypoint()
wpt.order = 10
wpt.latitude = 10
wpt.longitude = 100
wpt.altitude_msl = 0
wpt.save()
self.mission = MissionConfig()
self.mission.home_pos = pos
self.mission.emergent_last_known_pos = pos
self.mission.off_axis_odlc_pos = pos
self.mission.air_drop_pos = pos
self.mission.save()
self.mission.mission_waypoints.add(wpt)
self.mission.search_grid_points.add(wpt)
self.mission.save()
# Mission 2
self.mission2 = MissionConfig()
self.mission2.home_pos = pos
self.mission2.emergent_last_known_pos = pos
self.mission2.off_axis_odlc_pos = pos
self.mission2.air_drop_pos = pos
self.mission2.save()
self.mission2.mission_waypoints.add(wpt)
self.mission2.search_grid_points.add(wpt)
self.mission2.save()
def test_out_of_bounds(self):
"""Tests the UAS out of bounds method."""
(zone_details, uas_details) = TESTDATA_FLYZONE_EVALBOUNDS
# Create FlyZone objects
zones = []
for (alt_min, alt_max, wpts) in zone_details:
zone = FlyZone()
zone.altitude_msl_min = alt_min
zone.altitude_msl_max = alt_max
zone.save()
for wpt_id in range(len(wpts)):
(lat, lon) = wpts[wpt_id]
wpt = Waypoint()
wpt.order = wpt_id
wpt.latitude = lat
wpt.longitude = lon
wpt.altitude_msl = 0
wpt.save()
zone.boundary_pts.add(wpt)
zone.save()
zones.append(zone)
# For each user, validate time out of bounds
user_id = 0
epoch = timezone.now().replace(year=1970,
month=1,
day=1,
hour=0,
minute=0,
second=0,
microsecond=0)
for exp_violations, exp_out_of_bounds_time, log_details in uas_details:
# Create the logs
user = User.objects.create_user('testuser%d' % user_id,
'*****@*****.**', 'testpass')
user_id += 1
uas_logs = []
for (lat, lon, alt, timestamp) in log_details:
log = UasTelemetry()
log.user = user
log.latitude = lat
log.longitude = lon
log.altitude_msl = alt
log.uas_heading = 0
log.save()
log.timestamp = epoch + datetime.timedelta(seconds=timestamp)
log.save()
uas_logs.append(log)
# Assert out of bounds time matches expected
num_violations, out_of_bounds_time = \
FlyZone.out_of_bounds(zones, uas_logs)
self.assertEqual(num_violations, exp_violations)
self.assertAlmostEqual(out_of_bounds_time.total_seconds(),
exp_out_of_bounds_time)
def waypoints_from_data(self, waypoints_data):
"""Converts tuples of lat/lon/alt to a waypoint."""
waypoints = []
for i, waypoint in enumerate(waypoints_data):
(lat, lon, alt) = waypoint
wpt = Waypoint()
wpt.order = i
wpt.latitude = lat
wpt.longitude = lon
wpt.altitude_msl = alt
wpt.save()
waypoints.append(wpt)
return waypoints
def setUp(self):
super(TestTakeoffOrLandingEventModel, self).setUp()
# Mission
pos = GpsPosition()
pos.latitude = 10
pos.longitude = 100
pos.save()
apos = AerialPosition()
apos.latitude = 10
apos.longitude = 100
apos.altitude_msl = 1000
apos.save()
wpt = Waypoint()
wpt.latitude = 10
wpt.longitude = 100
wpt.altitude_msl = 1000
wpt.order = 10
wpt.save()
self.mission = MissionConfig()
self.mission.home_pos = pos
self.mission.lost_comms_pos = pos
self.mission.emergent_last_known_pos = pos
self.mission.off_axis_odlc_pos = pos
self.mission.map_center_pos = pos
self.mission.map_height_ft = 1
self.mission.air_drop_pos = pos
self.mission.ugv_drive_pos = pos
self.mission.save()
self.mission.mission_waypoints.add(wpt)
self.mission.search_grid_points.add(wpt)
self.mission.save()
# Mission 2
self.mission2 = MissionConfig()
self.mission2.home_pos = pos
self.mission2.lost_comms_pos = pos
self.mission2.emergent_last_known_pos = pos
self.mission2.off_axis_odlc_pos = pos
self.mission2.map_center_pos = pos
self.mission2.map_height_ft = 1
self.mission2.air_drop_pos = pos
self.mission2.ugv_drive_pos = pos
self.mission2.save()
self.mission2.mission_waypoints.add(wpt)
self.mission2.search_grid_points.add(wpt)
self.mission2.save()
self.ten_minutes = datetime.timedelta(minutes=10)
def setUp(self):
pos = GpsPosition()
pos.latitude = 10
pos.longitude = 100
pos.save()
wpt = Waypoint()
wpt.order = 10
wpt.latitude = 10
wpt.longitude = 100
wpt.altitude_msl = 1000
wpt.save()
config = MissionConfig()
config.home_pos = pos
config.lost_comms_pos = pos
config.emergent_last_known_pos = pos
config.off_axis_odlc_pos = pos
config.map_center_pos = pos
config.map_height_ft = 1
config.air_drop_pos = pos
config.ugv_drive_pos = pos
config.save()
config.mission_waypoints.add(wpt)
config.search_grid_points.add(wpt)
config.save()
user = User.objects.create_user('user', '*****@*****.**', 'pass')
self.feedback = MissionJudgeFeedback(
mission=config,
user=user,
flight_time=datetime.timedelta(seconds=1),
post_process_time=datetime.timedelta(seconds=2),
used_timeout=True,
min_auto_flight_time=True,
safety_pilot_takeovers=3,
out_of_bounds=6,
unsafe_out_of_bounds=7,
things_fell_off_uas=False,
crashed=False,
air_drop_accuracy=interop_admin_api_pb2.MissionJudgeFeedback.
WITHIN_05_FT,
ugv_drove_to_location=False,
operational_excellence_percent=9)
self.feedback.save()
def setUp(self):
# Mission
pos = GpsPosition()
pos.latitude = 10
pos.longitude = 100
pos.save()
wpt = Waypoint()
wpt.order = 10
wpt.latitude = 10
wpt.longitude = 100
wpt.altitude_msl = 0
wpt.save()
self.mission = MissionConfig()
self.mission.home_pos = pos
self.mission.lost_comms_pos = pos
self.mission.emergent_last_known_pos = pos
self.mission.off_axis_odlc_pos = pos
self.mission.map_center_pos = pos
self.mission.map_height_ft = 1
self.mission.air_drop_pos = pos
self.mission.ugv_drive_pos = pos
self.mission.save()
self.mission.mission_waypoints.add(wpt)
self.mission.search_grid_points.add(wpt)
self.mission.save()
# Mission 2
self.mission2 = MissionConfig()
self.mission2.home_pos = pos
self.mission2.lost_comms_pos = pos
self.mission2.emergent_last_known_pos = pos
self.mission2.off_axis_odlc_pos = pos
self.mission2.map_center_pos = pos
self.mission2.map_height_ft = 1
self.mission2.air_drop_pos = pos
self.mission2.ugv_drive_pos = pos
self.mission2.save()
self.mission2.mission_waypoints.add(wpt)
self.mission2.search_grid_points.add(wpt)
self.mission2.save()
self.user = User.objects.create_user('testuser', '*****@*****.**',
'testpass')
def setUp(self):
"""Setup the test case."""
super(TestOdlcEvaluator, self).setUp()
self.maxDiff = None
self.user = User.objects.create_user('user', '*****@*****.**',
'pass')
l1 = GpsPosition(latitude=38, longitude=-76)
l1.save()
l2 = GpsPosition(latitude=38.0003, longitude=-76)
l2.save()
l3 = GpsPosition(latitude=-38, longitude=76)
l3.save()
l4 = GpsPosition(latitude=0, longitude=0)
l4.save()
# Mission
pos = GpsPosition()
pos.latitude = 10
pos.longitude = 100
pos.save()
wpt = Waypoint()
wpt.order = 10
wpt.latitude = 10
wpt.longitude = 100
wpt.altitude_msl = 1000
wpt.save()
self.mission = MissionConfig()
self.mission.home_pos = pos
self.mission.lost_comms_pos = pos
self.mission.emergent_last_known_pos = pos
self.mission.off_axis_odlc_pos = pos
self.mission.map_center_pos = pos
self.mission.map_height_ft = 1
self.mission.air_drop_pos = pos
self.mission.ugv_drive_pos = pos
self.mission.save()
self.mission.mission_waypoints.add(wpt)
self.mission.search_grid_points.add(wpt)
self.mission.save()
event = TakeoffOrLandingEvent(user=self.user,
mission=self.mission,
uas_in_air=True)
event.save()
# A odlc worth full points.
self.submit1 = Odlc(mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.STANDARD,
location=l1,
orientation=interop_api_pb2.Odlc.S,
shape=interop_api_pb2.Odlc.SQUARE,
shape_color=interop_api_pb2.Odlc.WHITE,
alphanumeric='ABC',
alphanumeric_color=interop_api_pb2.Odlc.BLACK,
description='Submit test odlc 1',
description_approved=True,
autonomous=True,
thumbnail_approved=True)
self.submit1.save()
self.real1 = Odlc(mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.STANDARD,
location=l1,
orientation=interop_api_pb2.Odlc.S,
shape=interop_api_pb2.Odlc.SQUARE,
shape_color=interop_api_pb2.Odlc.WHITE,
alphanumeric='ABC',
alphanumeric_color=interop_api_pb2.Odlc.BLACK,
description='Real odlc 1')
self.real1.save()
# A odlc worth less than full points.
self.submit2 = Odlc(
mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.STANDARD,
location=l1,
orientation=interop_api_pb2.Odlc.N,
shape=interop_api_pb2.Odlc.CIRCLE,
shape_color=interop_api_pb2.Odlc.WHITE,
# alphanumeric set below
alphanumeric_color=interop_api_pb2.Odlc.BLACK,
description='Submit test odlc 2',
autonomous=False,
thumbnail_approved=True)
self.submit2.save()
self.real2 = Odlc(mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.STANDARD,
location=l2,
orientation=interop_api_pb2.Odlc.S,
shape=interop_api_pb2.Odlc.TRIANGLE,
shape_color=interop_api_pb2.Odlc.WHITE,
alphanumeric='ABC',
alphanumeric_color=interop_api_pb2.Odlc.BLACK,
description='Real test odlc 2')
self.real2.save()
# A odlc worth no points, so unmatched.
self.submit3 = Odlc(mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.STANDARD,
location=l4,
orientation=interop_api_pb2.Odlc.NW,
shape=interop_api_pb2.Odlc.PENTAGON,
shape_color=interop_api_pb2.Odlc.GRAY,
alphanumeric='XYZ',
alphanumeric_color=interop_api_pb2.Odlc.ORANGE,
description='Incorrect description',
autonomous=False,
thumbnail_approved=True)
self.submit3.save()
self.real3 = Odlc(
mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.STANDARD,
orientation=interop_api_pb2.Odlc.E,
shape=interop_api_pb2.Odlc.SEMICIRCLE,
shape_color=interop_api_pb2.Odlc.YELLOW,
alphanumeric='LMN',
# alphanumeric_color set below
location=l3,
description='Test odlc 3')
self.real3.save()
# Odlcs without approved image has no match value.
self.submit4 = Odlc(mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.EMERGENT,
location=l1,
description='Test odlc 4',
autonomous=False,
thumbnail_approved=False)
self.submit4.save()
self.real4 = Odlc(mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.EMERGENT,
location=l1,
description='Test odlc 4')
self.real4.save()
# A odlc without location worth fewer points.
self.submit5 = Odlc(mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.STANDARD,
orientation=interop_api_pb2.Odlc.N,
shape=interop_api_pb2.Odlc.TRAPEZOID,
shape_color=interop_api_pb2.Odlc.PURPLE,
alphanumeric='PQR',
alphanumeric_color=interop_api_pb2.Odlc.BLUE,
description='Test odlc 5',
autonomous=False,
thumbnail_approved=True)
self.submit5.save()
self.real5 = Odlc(mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.STANDARD,
location=l1,
orientation=interop_api_pb2.Odlc.N,
shape=interop_api_pb2.Odlc.TRAPEZOID,
shape_color=interop_api_pb2.Odlc.PURPLE,
alphanumeric='PQR',
alphanumeric_color=interop_api_pb2.Odlc.BLUE,
description='Test odlc 5')
self.real5.save()
# Emergent odlc with correct description.
self.submit6 = Odlc(mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.EMERGENT,
location=l1,
description='Submit test odlc 6',
description_approved=True,
autonomous=True,
thumbnail_approved=True)
self.submit6.save()
self.real6 = Odlc(mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.EMERGENT,
location=l1,
description='Real odlc 1',
description_approved=True)
self.real6.save()
event = TakeoffOrLandingEvent(user=self.user,
mission=self.mission,
uas_in_air=False)
event.save()
# submit2 updated after landing.
self.submit2.alphanumeric = 'ABC'
self.submit2.update_last_modified()
self.submit2.save()
self.submit2.update_last_modified()
self.submit3.alphanumeric_color = interop_api_pb2.Odlc.YELLOW
self.submit3.update_last_modified()
self.submit3.save()
# Unused but not unmatched odlc.
self.submit7 = Odlc(mission=self.mission,
user=self.user,
odlc_type=interop_api_pb2.Odlc.STANDARD,
location=l4,
alphanumeric_color=interop_api_pb2.Odlc.BLACK,
description='Submit unused test odlc 1',
autonomous=False,
thumbnail_approved=True)
self.submit7.save()
self.submitted_odlcs = [
self.submit7, self.submit6, self.submit5, self.submit4,
self.submit3, self.submit2, self.submit1
]
self.real_odlcs = [
self.real1, self.real2, self.real3, self.real4, self.real5,
self.real6
]
self.flights = TakeoffOrLandingEvent.flights(self.mission, self.user)
