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