def test_taf_line(self): """ Tests converting TAF line data into into a single spoken string """ units = structs.Units(**static.NA_UNITS) line = { 'altimeter': core.make_number('2992'), 'clouds': [core.make_cloud('BKN015CB')], 'end_time': core.make_timestamp('1206'), 'icing': ['611005'], 'other': ['+RA'], 'start_time': core.make_timestamp('1202'), 'turbulance': ['540553'], 'type': 'FROM', 'visibility': core.make_number('3'), 'wind_direction': core.make_number('360'), 'wind_gust': core.make_number('20'), 'wind_shear': 'WS020/07040KT', 'wind_speed': core.make_number('12'), } line.update({ k: None for k in ('flight_rules', 'probability', 'raw', 'sanitized') }) line = structs.TafLineData(**line) spoken = ( 'From 2 to 6 zulu, Winds three six zero at 12kt gusting to 20kt. ' 'Wind shear 2000inHg from zero seven zero at 40kt. Visibility three miles. ' 'Altimeter two nine point nine two. Heavy Rain. ' 'Broken layer at 1500ft (Cumulonimbus). ' 'Occasional moderate turbulence in clouds from 5500ft to 8500ft. ' 'Light icing from 10000ft to 15000ft') ret = speech.taf_line(line, units) self.assertIsInstance(ret, str) self.assertEqual(ret, spoken)
def test_get_wind(self): """ Tests that the wind item gets removed and split into its components """ # Both us knots as the default unit, so just test North American default for wx, unit, *wind, varv in ((['1' ], 'kt', (None, ), (None, ), (None, ), []), (['12345(E)', 'G50', '1'], 'kt', ('123', 123), ('45', 45), ('50', 50), []), (['O1234G56', '1'], 'kt', ('012', 12), ('34', 34), ('56', 56), []), (['36010KTS', 'G20', '300V060', '1'], 'kt', ('360', 360), ('10', 10), ('20', 20), [ ('300', 300), ('060', 60) ]), (['VRB10MPS', '1'], 'm/s', ('VRB', ), ('10', 10), (None, ), []), (['VRB20G30KMH', '1'], 'km/h', ('VRB', ), ('20', 20), ('30', 30), [])): units = structs.Units(**static.NA_UNITS) wx, *winds, var = core.get_wind(wx, units) self.assertEqual(wx, ['1']) for i in range(len(wind)): self.assert_number(winds[i], *wind[i]) if varv: self.assertIsInstance(varv, list) for i in range(2): self.assert_number(var[i], *varv[i]) self.assertEqual(units.wind_speed, unit)
def test_taf(self): """ Tests end-to-end TAF translation """ units = structs.Units(**static.NA_UNITS) line_data = { 'altimeter': core.make_number('2992'), 'clouds': [core.make_cloud('BKN015CB')], 'icing': ['611005'], 'other': ['+RA'], 'turbulance': ['540553'], 'visibility': core.make_number('3'), 'wind_direction': core.make_number('360'), 'wind_gust': core.make_number('20'), 'wind_shear': 'WS020/07040KT', 'wind_speed': core.make_number('12') } line_data.update({ k: '' for k in ('raw', 'end_time', 'start_time', 'probability', 'type', 'flight_rules', 'sanitized') }) data = {'max_temp': 'TX20/1518Z', 'min_temp': 'TN00/00', 'remarks': ''} data.update({ k: '' for k in ('raw', 'station', 'time', 'start_time', 'end_time') }) data = structs.TafData(forecast=[structs.TafLineData(**line_data)], **data) # noinspection PyArgumentList line_trans = structs.TafLineTrans( altimeter='29.92 inHg (1013 hPa)', clouds='Broken layer at 1500ft (Cumulonimbus) - Reported AGL', icing='Light icing from 10000ft to 15000ft', other='Heavy Rain', turbulance= 'Occasional moderate turbulence in clouds from 5500ft to 8500ft', visibility='3sm (4.8km)', wind_shear='Wind shear 2000ft from 070 at 40kt', wind='N-360 at 12kt gusting to 20kt') trans = structs.TafTrans( forecast=[line_trans], max_temp='Maximum temperature of 20°C (68°F) at 15-18:00Z', min_temp='Minimum temperature of 0°C (32°F) at 00:00Z', remarks={}) translated = translate.taf(data, units) self.assertIsInstance(translated, structs.TafTrans) for line in translated.forecast: self.assertIsInstance(line, structs.TafLineTrans) self.assertEqual(translated, trans)
def test_taf(self): """ Tests converting a TafData report into a single spoken string """ units = structs.Units(**static.NA_UNITS) # noinspection PyUnresolvedReferences empty_line = { k: None for k in structs.TafLineData.__dataclass_fields__.keys() } forecast = [ structs.TafLineData(**{ **empty_line, **line }) for line in ( { 'type': 'FROM', 'start_time': core.make_timestamp('0410Z'), 'end_time': core.make_timestamp('0414Z'), 'visibility': core.make_number('3'), 'wind_direction': core.make_number('360'), 'wind_gust': core.make_number('20'), 'wind_speed': core.make_number('12'), }, { 'type': 'PROB', 'probability': core.make_number('45'), 'start_time': core.make_timestamp('0412Z'), 'end_time': core.make_timestamp('0414Z'), 'visibility': core.make_number('M1/4'), }, ) ] # noinspection PyArgumentList taf = structs.TafData(raw=None, remarks=None, station=None, time=None, forecast=forecast, start_time=core.make_timestamp('0410Z'), end_time=core.make_timestamp('0414Z')) ret = speech.taf(taf, units) spoken = ( "Starting on .* 4th - From 10 to 14 zulu, " "Winds three six zero at 12kt gusting to 20kt. Visibility three miles. " r"From 12 to 14 zulu, there's a 45% chance for Visibility " "less than one quarter of a mile") self.assertIsInstance(ret, str) assert re.match(spoken, ret)
def test_get_altimeter(self): """ Tests that the correct alimeter item gets removed from the end of the wx list """ # North American default units = structs.Units(**static.NA_UNITS) for wx, alt in ( (['1', '2'], (None, )), (['1', '2', 'A2992'], ('2992', 29.92)), (['1', '2', '2992'], ('2992', 29.92)), (['1', '2', 'A2992', 'Q1000'], ('2992', 29.92)), (['1', '2', 'Q1000', 'A2992'], ('2992', 29.92)), (['1', '2', 'Q1000'], ('1000', 1000)), ): self.assertEqual(units.altimeter, 'inHg') retwx, ret_alt = core.get_altimeter(wx, units) self.assertEqual(retwx, ['1', '2']) self.assert_number(ret_alt, *alt) # The last one should have changed the unit self.assertEqual(units.altimeter, 'hPa') # International units = structs.Units(**static.IN_UNITS) for wx, alt in ( (['1', '2'], (None, )), (['1', '2', 'Q.1000'], ('1000', 1000)), (['1', '2', 'Q1000/10'], ('1000', 1000)), (['1', '2', 'A2992', 'Q1000'], ('1000', 1000)), (['1', '2', 'Q1000', 'A2992'], ('1000', 1000)), (['1', '2', 'A2992'], ('2992', 29.92)), ): self.assertEqual(units.altimeter, 'hPa') retwx, ret_alt = core.get_altimeter(wx, units, 'IN') self.assertEqual(retwx, ['1', '2']) self.assert_number(ret_alt, *alt) # The last one should have changed the unit self.assertEqual(units.altimeter, 'inHg')
def test_shared(): units = structs.Units(**static.NA_UNITS) data = structs.SharedData(altimeter=core.make_number('2992'), clouds=[core.make_cloud('OVC060')], flight_rules='', other=['RA'], sanitized='', visibility=core.make_number('10'), wind_direction=core.make_number('0'), wind_gust=core.make_number('0'), wind_speed=core.make_number('0')) # noinspection PyTypeChecker trans = translate.shared(data, units) assert isinstance(trans, dict) for key in ('altimeter', 'clouds', 'other', 'visibility'): assert key in trans assert trans[key]
def test_metar(self): """ Tests end-to-end METAR translation """ units = structs.Units(**static.NA_UNITS) data = { 'altimeter': core.make_number('2992'), 'clouds': [core.make_cloud('BKN015CB')], 'dewpoint': core.make_number('M01'), 'other': ['+RA'], 'temperature': core.make_number('03'), 'visibility': core.make_number('3'), 'wind_direction': core.make_number('360'), 'wind_gust': core.make_number('20'), 'wind_speed': core.make_number('12'), 'wind_variable_direction': [core.make_number('340'), core.make_number('020')] } data.update({ k: '' for k in ('raw', 'remarks', 'station', 'time', 'flight_rules', 'remarks_info', 'runway_visibility', 'sanitized') }) data = structs.MetarData(**data) # noinspection PyArgumentList trans = structs.MetarTrans( altimeter='29.92 inHg (1013 hPa)', clouds='Broken layer at 1500ft (Cumulonimbus) - Reported AGL', dewpoint='-1°C (30°F)', other='Heavy Rain', remarks={}, temperature='3°C (37°F)', visibility='3sm (4.8km)', wind='N-360 (variable 340 to 020) at 12kt gusting to 20kt') translated = translate.metar(data, units) self.assertIsInstance(translated, structs.MetarTrans) self.assertEqual(translated, trans)
def test_metar(self): """ Tests converting METAR data into into a single spoken string """ units = structs.Units(**static.NA_UNITS) data = { 'altimeter': core.make_number('2992'), 'clouds': [core.make_cloud('BKN015CB')], 'dewpoint': core.make_number('M01'), 'other': ['+RA'], 'temperature': core.make_number('03'), 'visibility': core.make_number('3'), 'wind_direction': core.make_number('360'), 'wind_gust': core.make_number('20'), 'wind_speed': core.make_number('12'), 'wind_variable_direction': [core.make_number('340'), core.make_number('020', speak='020')] } data.update({ k: None for k in ('raw', 'remarks', 'station', 'time', 'flight_rules', 'remarks_info', 'runway_visibility', 'sanitized') }) data = structs.MetarData(**data) spoken = ( 'Winds three six zero (variable three four zero to zero two zero) ' 'at 12kt gusting to 20kt. Visibility three miles. ' 'Temperature three degrees Celsius. Dew point minus one degree Celsius. ' 'Altimeter two nine point nine two. Heavy Rain. ' 'Broken layer at 1500ft (Cumulonimbus)') ret = speech.metar(data, units) self.assertIsInstance(ret, str) self.assertEqual(ret, spoken)
def test_get_visibility(self): """ Tests that the visibility item(s) gets removed and cleaned """ for wx, unit, visibility in ( (['1'], 'sm', (None, )), (['05SM', '1'], 'sm', ('5', 5)), (['10SM', '1'], 'sm', ('10', 10)), (['P6SM', '1'], 'sm', ('P6', )), (['M1/4SM', '1'], 'sm', ('M1/4', )), (['1/2SM', '1'], 'sm', ('1/2', 0.5)), (['2', '1/2SM', '1'], 'sm', ('5/2', 2.5)), (['1000', '1'], 'm', ('1000', 1000)), (['1000E', '1'], 'm', ('1000', 1000)), (['1000NDV', '1'], 'm', ('1000', 1000)), (['M1000', '1'], 'm', ('1000', 1000)), (['2KM', '1'], 'm', ('2000', 2000)), ): units = structs.Units(**static.NA_UNITS) wx, vis = core.get_visibility(wx, units) self.assertEqual(wx, ['1']) self.assert_number(vis, *visibility) self.assertEqual(units.visibility, unit)