def test_read_write(self):
for trace in self.test_traces:
self.tracebuf2.write(trace)
read_traces = self.tracebuf2.read()
# Using dicts to make it easier to evaluate equality.
test_dict = [Tracebuf2Message.toDict(t) for t in self.test_traces]
read_dict = [Tracebuf2Message.toDict(t) for t in read_traces]
# Asserting that what was read it the same as what was written.
self.assertTrue(test_dict == read_dict)
def setUp(self):
self.tracebuf2 = Tracebuf2Ring('WAVE', 'MOD_WILDCARD')
dict_trace = dict(
pinno=0, nsamp=100, starttime=0.0, endtime=0.0, samprate=100,
sta='BRU2', net='PA', chan='HHZ', loc='--', version='20',
datatype='i4', quality='00', pad='', samples=[0] * 100)
self.test_traces = []
self.test_other_traces = []
for i in range(100):
# Preparing sets of messages to write.
dict_trace.update(dict(starttime=i, endtime=i+1, sta='BRU2'))
self.test_traces.append(Tracebuf2Message.fromDict(dict_trace))
dict_trace.update(dict(sta='BRU3'))
self.test_other_traces.append(Tracebuf2Message.fromDict(dict_trace))
