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))