def setUp(self):
self.reader = StringIterator(TEST_TAGREADER)
class TestTagReader(unittest.TestCase):
def setUp(self):
self.reader = StringIterator(TEST_TAGREADER)
def test_next(self):
self.assertEqual((0, 'SECTION'), next(self.reader))
def test_undo_last(self):
self.reader.__next__()
self.reader.undotag()
self.assertEqual((0, 'SECTION'), next(self.reader))
def test_error_on_multiple_undo_last(self):
next(self.reader)
self.reader.undotag()
with self.assertRaises(ValueError):
self.reader.undotag()
def test_error_undo_last_before_first_read(self):
with self.assertRaises(ValueError):
self.reader.undotag()
def test_lineno(self):
next(self.reader)
self.assertEqual(2, self.reader.lineno)
def test_lineno_with_undo(self):
next(self.reader)
self.reader.undotag()
self.assertEqual(0, self.reader.lineno)
def test_lineno_with_undo_next(self):
next(self.reader)
self.reader.undotag()
next(self.reader)
self.assertEqual(2, self.reader.lineno)
def test_to_list(self):
tags = list(self.reader)
self.assertEqual(8, len(tags))
def test_undo_eof(self):
for tag in self.reader:
if tag == (0, 'EOF'):
self.reader.undotag()
break
tag = next(self.reader)
self.assertEqual((0, 'EOF'), tag)
with self.assertRaises(StopIteration):
self.reader.__next__()
def test_no_eof(self):
tags = list(StringIterator(TEST_NO_EOF))
self.assertEqual(7, len(tags))
self.assertEqual((0, 'ENDSEC'), tags[-1])
def test_strtag_int(self):
self.assertEqual(' 1\n1\n', strtag((1, 1)))
def test_strtag_float(self):
self.assertEqual(' 10\n3.1415\n', strtag((10, 3.1415)))
def test_strtag_str(self):
self.assertEqual(' 0\nSECTION\n', strtag((0, 'SECTION')))
def setUp(self):
self.reader = StringIterator(TEST_TAGREADER)
def test_no_eof(self):
tags = list(StringIterator(TEST_NO_EOF))
self.assertEqual(7, len(tags))
self.assertEqual((0, 'ENDSEC'), tags[-1])
class TestTagReader(unittest.TestCase):
def setUp(self):
self.reader = StringIterator(TEST_TAGREADER)
def test_next(self):
self.assertEqual((0, 'SECTION'), next(self.reader))
def test_undo_last(self):
self.reader.__next__()
self.reader.undotag()
self.assertEqual((0, 'SECTION'), next(self.reader))
def test_error_on_multiple_undo_last(self):
next(self.reader)
self.reader.undotag()
with self.assertRaises(ValueError):
self.reader.undotag()
def test_error_undo_last_before_first_read(self):
with self.assertRaises(ValueError):
self.reader.undotag()
def test_lineno(self):
next(self.reader)
self.assertEqual(2, self.reader.lineno)
def test_lineno_with_undo(self):
next(self.reader)
self.reader.undotag()
self.assertEqual(0, self.reader.lineno)
def test_lineno_with_undo_next(self):
next(self.reader)
self.reader.undotag()
next(self.reader)
self.assertEqual(2, self.reader.lineno)
def test_to_list(self):
tags = list(self.reader)
self.assertEqual(8, len(tags))
def test_undo_eof(self):
for tag in self.reader:
if tag == (0, 'EOF'):
self.reader.undotag()
break
tag = next(self.reader)
self.assertEqual((0, 'EOF'), tag)
with self.assertRaises(StopIteration):
self.reader.__next__()
def test_no_eof(self):
tags = list(StringIterator(TEST_NO_EOF))
self.assertEqual(7, len(tags))
self.assertEqual((0, 'ENDSEC'), tags[-1])
def test_strtag_int(self):
self.assertEqual(' 1\n1\n', strtag( (1,1) ))
def test_strtag_float(self):
self.assertEqual(' 10\n3.1415\n', strtag( (10, 3.1415) ))
def test_strtag_str(self):
self.assertEqual(' 0\nSECTION\n', strtag( (0, 'SECTION') ))
