def assertExplorerContents(self,
explorer,
count_unfiltered=3,
count_filtered=1):
self.assertCountEqual(explorer.all_stations, [
self._station("synop", 1, 12.34560, 76.54320, None),
self._station("amdar", 2, 12.34560, 76.54320, "foo"),
])
self.assertCountEqual(explorer.stations, [
self._station("amdar", 2, 12.34560, 76.54320, "foo"),
])
self.assertCountEqual(explorer.all_reports, ["amdar", "synop"])
self.assertEqual(explorer.reports, ["amdar"])
self.assertEqual(explorer.all_levels, [(10, 11, 15, 22)])
self.assertEqual(explorer.levels, [(10, 11, 15, 22)])
self.assertEqual(explorer.all_tranges, [(20, 111, 222),
(20, 111, 223)])
self.assertEqual(explorer.tranges, [(20, 111, 223)])
self.assertEqual(explorer.all_varcodes, ["B01011", "B01012"])
self.assertEqual(explorer.varcodes, ["B01012"])
self.assertEqual(
explorer.all_stats,
dballe.ExplorerStats((datetime.datetime(1945, 4, 25, 8, 0),
datetime.datetime(1945, 4, 25, 12,
0), count_unfiltered)))
self.assertEqual(
explorer.stats,
dballe.ExplorerStats(
(datetime.datetime(1945, 4, 25, 12,
0), datetime.datetime(1945, 4, 25, 12,
0), count_filtered)))
def test_create_from_msgs(self):
importer = dballe.Importer("BUFR")
explorer = self._explorer()
with explorer.rebuild() as update:
with importer.from_file(test_pathname("bufr/gts-acars-uk1.bufr")) as imp:
update.add_messages(imp)
explorer.set_filter({"level": dballe.Level(102, 6260000)})
self.assertCountEqual(explorer.all_stations, [
self._station("amdar", None, 48.90500, 10.63667, "EU3375"),
])
self.assertCountEqual(explorer.stations, [
self._station("amdar", None, 48.90500, 10.63667, "EU3375"),
])
self.assertEqual(explorer.all_reports, ["amdar"])
self.assertEqual(explorer.reports, ["amdar"])
self.assertEqual(explorer.all_levels, [dballe.Level(102, 6260000), None])
self.assertEqual(explorer.levels, [dballe.Level(102, 6260000)])
self.assertEqual(explorer.all_tranges, [dballe.Trange(254, 0, 0), None])
self.assertEqual(explorer.tranges, [dballe.Trange(254, 0, 0)])
self.assertCountEqual(explorer.all_varcodes, [
"B04001", "B04002", "B04003", "B04004", "B04005", "B01011", "B05001", "B06001",
"B01006", "B02061", "B02062", "B02064", "B07030", "B08004", "B11001", "B11002", "B12101", "B13002",
])
self.assertCountEqual(explorer.varcodes, [
"B01006", "B02061", "B02062", "B02064", "B07030", "B08004", "B11001", "B11002", "B12101", "B13002",
])
self.assertEqual(explorer.all_stats, dballe.ExplorerStats((
datetime.datetime(2009, 2, 24, 11, 31), datetime.datetime(2009, 2, 24, 11, 31), 18)))
self.assertEqual(explorer.stats, dballe.ExplorerStats((
datetime.datetime(2009, 2, 24, 11, 31), datetime.datetime(2009, 2, 24, 11, 31), 10)))