def test_stream():
inventory = get_inventory()
channels = ["HN1", "HN2", "HNZ"]
data = np.random.rand(1000)
traces = []
network = inventory.networks[0]
station = network.stations[0]
chlist = station.channels
channelcodes = [ch.code for ch in chlist]
for channel in channels:
chidx = channelcodes.index(channel)
channeldata = chlist[chidx]
header = {
"sampling_rate": channeldata.sample_rate,
"npts": len(data),
"network": network.code,
"location": channeldata.location_code,
"station": station.code,
"channel": channel,
"starttime": UTCDateTime(2010, 1, 1, 0, 0, 0),
}
trace = Trace(data=data, header=header)
traces.append(trace)
invstream = StationStream(traces=traces, inventory=inventory)
inventory2 = invstream.getInventory()
inv2_channel1 = inventory2.networks[0].stations[0].channels[0]
inv_channel1 = inventory2.networks[0].stations[0].channels[0]
assert inv_channel1.code == inv2_channel1.code
# test the streamparam functionality
statsdict = {"name": "Fred", "age": 34}
invstream.setStreamParam("stats", statsdict)
assert invstream.getStreamParamKeys() == ["stats"]
cmpdict = invstream.getStreamParam("stats")
assert statsdict == cmpdict
def test_stream():
inventory = get_inventory()
channels = ['HN1', 'HN2', 'HNZ']
data = np.random.rand(1000)
traces = []
network = inventory.networks[0]
station = network.stations[0]
chlist = station.channels
channelcodes = [ch.code for ch in chlist]
for channel in channels:
chidx = channelcodes.index(channel)
channeldata = chlist[chidx]
header = {
'sampling_rate': channeldata.sample_rate,
'npts': len(data),
'network': network.code,
'location': channeldata.location_code,
'station': station.code,
'channel': channel,
'starttime': UTCDateTime(2010, 1, 1, 0, 0, 0)
}
trace = Trace(data=data, header=header)
traces.append(trace)
invstream = StationStream(traces=traces, inventory=inventory)
inventory2 = invstream.getInventory()
inv2_channel1 = inventory2.networks[0].stations[0].channels[0]
inv_channel1 = inventory2.networks[0].stations[0].channels[0]
assert inv_channel1.code == inv2_channel1.code
# test the streamparam functionality
statsdict = {'name': 'Fred', 'age': 34}
invstream.setStreamParam('stats', statsdict)
assert invstream.getStreamParamKeys() == ['stats']
cmpdict = invstream.getStreamParam('stats')
assert statsdict == cmpdict