class DocumentTest(unittest.TestCase):
def setUp(self):
self.document = Document(
'line 1\n' +
'line 2\n' +
'line 3\n' +
'line 4\n',
len('line 1\n' + 'lin')
)
def test_current_char(self):
self.assertEqual(self.document.current_char, 'e')
def test_text_before_cursor(self):
self.assertEqual(self.document.text_before_cursor, 'line 1\nlin')
def test_text_after_cursor(self):
self.assertEqual(self.document.text_after_cursor,
'e 2\n' +
'line 3\n' +
'line 4\n')
def test_lines(self):
self.assertEqual(self.document.lines, [
'line 1',
'line 2',
'line 3',
'line 4', ''])
def test_line_count(self):
self.assertEqual(self.document.line_count, 5)
def test_current_line_before_cursor(self):
self.assertEqual(self.document.current_line_before_cursor, 'lin')
def test_current_line_after_cursor(self):
self.assertEqual(self.document.current_line_after_cursor, 'e 2')
def test_current_line(self):
self.assertEqual(self.document.current_line, 'line 2')
def test_cursor_position(self):
self.assertEqual(self.document.cursor_position_row, 1)
self.assertEqual(self.document.cursor_position_col, 3)
d = Document('', 0)
self.assertEqual(d.cursor_position_row, 0)
self.assertEqual(d.cursor_position_col, 0)
def test_translate_index_to_position(self):
pos = self.document.translate_index_to_position(
len('line 1\nline 2\nlin'))
self.assertEqual(pos[0], 2)
self.assertEqual(pos[1], 3)
pos = self.document.translate_index_to_position(0)
self.assertEqual(pos, (0, 0))
def _get_positions_to_highlight(self, document):
"""
Return a list of (row, col) tuples that need to be highlighted.
"""
# Try for the character under the cursor.
if document.current_char and document.current_char in self.chars:
pos = document.find_matching_bracket_position(
start_pos=document.cursor_position - self.max_cursor_distance,
end_pos=document.cursor_position + self.max_cursor_distance)
# Try for the character before the cursor.
elif (document.char_before_cursor
and document.char_before_cursor in self._closing_braces
and document.char_before_cursor in self.chars):
document = Document(document.text, document.cursor_position - 1)
pos = document.find_matching_bracket_position(
start_pos=document.cursor_position - self.max_cursor_distance,
end_pos=document.cursor_position + self.max_cursor_distance)
else:
pos = None
# Return a list of (row, col) tuples that need to be highlighted.
if pos:
pos += document.cursor_position # pos is relative.
row, col = document.translate_index_to_position(pos)
return [(row, col),
(document.cursor_position_row,
document.cursor_position_col)]
else:
return []
def _get_positions_to_highlight(self, document):
"""
Return a list of (row, col) tuples that need to be highlighted.
"""
# Try for the character under the cursor.
if document.current_char and document.current_char in self.chars:
pos = document.find_matching_bracket_position(
start_pos=document.cursor_position - self.max_cursor_distance,
end_pos=document.cursor_position + self.max_cursor_distance)
# Try for the character before the cursor.
elif (document.char_before_cursor and document.char_before_cursor in
self._closing_braces and document.char_before_cursor in self.chars):
document = Document(document.text, document.cursor_position - 1)
pos = document.find_matching_bracket_position(
start_pos=document.cursor_position - self.max_cursor_distance,
end_pos=document.cursor_position + self.max_cursor_distance)
else:
pos = None
# Return a list of (row, col) tuples that need to be highlighted.
if pos:
pos += document.cursor_position # pos is relative.
row, col = document.translate_index_to_position(pos)
return [(row, col), (document.cursor_position_row, document.cursor_position_col)]
else:
return []
