def test_lookup_lineno(self): offsets = _get_line_offsets("12\n3\n4") self.assertEqual(_lookup_lineno(0, offsets), 1) self.assertEqual(_lookup_lineno(1, offsets), 1) self.assertEqual(_lookup_lineno(2, offsets), 1) self.assertEqual(_lookup_lineno(3, offsets), 2) self.assertEqual(_lookup_lineno(4, offsets), 2) self.assertEqual(_lookup_lineno(5, offsets), 3)