def test__get_interval_code(): """edge_test.MiniSeedFactory_test.test__get_interval_code()""" assert_equal(MiniSeedFactory()._get_interval_code("day"), "P") assert_equal(MiniSeedFactory()._get_interval_code("hour"), "R") assert_equal(MiniSeedFactory()._get_interval_code("minute"), "U") assert_equal(MiniSeedFactory()._get_interval_code("second"), "L") assert_equal(MiniSeedFactory()._get_interval_code("tenhertz"), "B")
def test__get_edge_location(): """edge_test.MiniSeedFactory_test.test__get_edge_location() """ # Call _get_edge_location, make certain it returns the correct edge # location code. assert_equal(MiniSeedFactory()._get_edge_location("", "", "variation", ""), "R0") assert_equal( MiniSeedFactory()._get_edge_location("", "", "quasi-definitive", ""), "Q0" ) assert_equal(MiniSeedFactory()._get_edge_location("", "", "definitive", ""), "D0")
def test__set_metadata(): """edge_test.MiniSeedFactory_test.test__set_metadata()""" # Call _set_metadata with 2 traces, and make certain the stats get # set for both traces. trace1 = Trace() trace2 = Trace() stream = Stream(traces=[trace1, trace2]) MiniSeedFactory()._set_metadata(stream, "BOU", "H", "variation", "minute") assert_equal(stream[0].stats["channel"], "H") assert_equal(stream[1].stats["channel"], "H")
def test__get_edge_channel(): """edge_test.MiniSeedFactory_test.test__get_edge_channel()""" # Call private function _get_edge_channel, make certain # it gets back the appropriate 2 character code. factory = MiniSeedFactory() assert_equal(factory._get_edge_channel("", "D", "", "minute"), "UFD") assert_equal(factory._get_edge_channel("", "U", "", "minute"), "UFU") assert_equal(factory._get_edge_channel("", "F", "", "minute"), "UFF") assert_equal(factory._get_edge_channel("", "H", "", "minute"), "UFH") assert_equal(factory._get_edge_channel("", "BEU", "", "minute"), "BEU") assert_equal(factory._get_edge_channel("", "Dst4", "", "minute"), "UX4") assert_equal(factory._get_edge_channel("", "Dst3", "", "minute"), "UX3") assert_equal(factory._get_edge_channel("", "E-E", "", "minute"), "UQE") assert_equal(factory._get_edge_channel("", "E-N", "", "minute"), "UQN")
def test__put_timeseries(): """edge_test.MiniSeedFactory_test.test__put_timeseries()""" trace1 = __create_trace([0, 1, 2, 3, numpy.nan, 5, 6, 7, 8, 9], channel="H") client = MockMiniSeedInputClient() factory = MiniSeedFactory() factory.write_client = client factory.put_timeseries(Stream(trace1), channels=("H")) # put timeseries should call close when done assert_equal(client.close_called, True) # trace should be split in 2 blocks at gap sent = client.last_sent assert_equal(len(sent), 2) # first trace includes [0...4] assert_equal(sent[0].stats.channel, "LFH") assert_equal(len(sent[0]), 4) assert_equal(sent[0].stats.endtime, trace1.stats.starttime + 3) # second trace includes [5...9] assert_equal(sent[1].stats.channel, "LFH") assert_equal(len(sent[1]), 5) assert_equal(sent[1].stats.starttime, trace1.stats.starttime + 5) assert_equal(sent[1].stats.endtime, trace1.stats.endtime)
def dont_get_timeseries(): """edge_test.MiniSeedFactory_test.test_get_timeseries()""" # Call get_timeseries, and test stats for comfirmation that it came back. # TODO, need to pass in host and port from a config file, or manually # change for a single test. edge_factory = MiniSeedFactory(host="TODO", port="TODO") timeseries = edge_factory.get_timeseries( UTCDateTime(2015, 3, 1, 0, 0, 0), UTCDateTime(2015, 3, 1, 1, 0, 0), "BOU", ("H"), "variation", "minute", ) assert_equal( timeseries.select(channel="H")[0].stats.station, "BOU", "Expect timeseries to have stats", ) assert_equal( timeseries.select(channel="H")[0].stats.channel, "H", "Expect timeseries stats channel to be equal to H", )
def test__get_edge_network(): """edge_test.MiniSeedFactory_test.test__get_edge_network()""" # _get_edge_network should always return NT for use by USGS geomag assert_equal(MiniSeedFactory()._get_edge_network(" ", " ", " ", " "), "NT")
def test__get_edge_station(): """edge_test.MiniSeedFactory_test.test__get_edge_station()""" # _get_edge_station will return the observatory code passed in. assert_equal(MiniSeedFactory()._get_edge_station("BOU", " ", " ", " "), "BOU")