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 create_test_file(): with warnings.catch_warnings(): warnings.simplefilter('ignore', RuntimeWarning) filename = os.tempnam() with MWKStream._create_file(filename) as fp: with open('tests/codec.dat') as codec_fp: fp._write([0, 0, pickle.load(codec_fp)]) tagmap = dict((item[0], i + 2) for i, item in enumerate(test_data)) fp._write([1, 1, tagmap]) code = 2 for tag, data in test_data: fp._write([code, code, data]) code += 1 return filename
def create_test_file(): with warnings.catch_warnings(): warnings.simplefilter('ignore', RuntimeWarning) filename = os.tempnam() with MWKStream._create_file(filename) as fp: with open('tests/codec.dat') as codec_fp: fp._write([0, 0, pickle.load(codec_fp)]) tagmap = dict((item[0], i+2) for i, item in enumerate(test_data)) fp._write([1, 1, tagmap]) code = 2 for tag, data in test_data: fp._write([code, code, data]) code += 1 return filename
def setUpClass(cls): cls.filename, cls.outstream = cls.create_file() cls.instream = MWKStream.open_file(cls.filename)
def create_file(): with warnings.catch_warnings(): warnings.simplefilter("ignore", RuntimeWarning) filename = os.tempnam() fp = MWKStream._create_file(filename) return filename, fp
def create_file(self, events=()): with MWKStream._create_file(self.filename) as fp: for evt in events: fp._write(evt)
def create_file(): with warnings.catch_warnings(): warnings.simplefilter('ignore', RuntimeWarning) filename = os.tempnam() fp = MWKStream._create_file(filename) return filename, fp