def test_iteration_and_context_management(self): evts = [[1, 11, "one"], [2, 22, "two"], [3, 33, "three"], [4, 44, "four"]] filename, outstream = self.create_file() with outstream: for e in evts: outstream._write(e) # Write to closed file should fail self.assertRaises(IOError, outstream._write, evts[0]) with MWKStream.open_file(filename) as instream: # Attempt to open already-open stream should fail self.assertRaises(IOError, instream.open) for i, e in enumerate(instream): self.assertEvent(e, *evts[i]) self.assertEqual(len(evts) - 1, i) # Once we've hit EOF, read_event should return None self.assertIs(instream.read_event(), None) self.assertIs(instream.read_event(), None) # Read from closed file should fail self.assertRaises(IOError, instream._read)
def test_iteration_and_context_management(self): evts = [ [1, 11, 'one'], [2, 22, 'two'], [3, 33, 'three'], [4, 44, 'four'], ] filename, outstream = self.create_file() with outstream: for e in evts: outstream._write(e) # Write to closed file should fail self.assertRaises(IOError, outstream._write, evts[0]) with MWKStream.open_file(filename) as instream: # Attempt to open already-open stream should fail self.assertRaises(IOError, instream.open) for i, e in enumerate(instream): self.assertEvent(e, *evts[i]) self.assertEqual(len(evts) - 1, i) # Once we've hit EOF, read_event should return None self.assertIs(instream.read_event(), None) self.assertIs(instream.read_event(), None) # Read from closed file should fail self.assertRaises(IOError, instream._read)
def setUpClass(cls): cls.event_counts = defaultdict(lambda: 0) cls.event_times = [] with MWKStream.open_file(cls.filename) as stream: for evt in stream: cls.event_counts[evt.code] += 1 cls.event_times.append(evt.time) cls.event_counts = dict(cls.event_counts) cls.event_times = numpy.array(cls.event_times) cls.fp = MWKFile(cls.filename)
def setUpClass(cls): cls.filename, cls.outstream = cls.create_file() cls.instream = MWKStream.open_file(cls.filename)