class ReadCurrentClassEof(unittest.TestCase): def init_reader(self, s): from onyx.util.stream import Stream from onyx.reader import Reader stream = Stream.from_sequence(s) self.reader = Reader(stream) def assert_reader_at_end(self): if not self.reader.is_at_end(): self.fail("reader is not at end") def setUp(self): self.init_reader("") def runTest(self): """reading at eof should produce the proper classification""" self.assertEqual({"eof"}, self.reader.current_class()) self.assert_reader_at_end()