def test_get_highlighted_cells(self):
    """Tests whether the references are returned as expected."""
    table = self._get_toy_table()
    cell_indices = [[0, 0], [1, 1]]
    subtable = table_to_text_utils.get_highlighted_subtable(table, cell_indices)
    assert len(subtable) == 2
    assert subtable[0] == table[0][0]
    assert subtable[1] == table[1][1]
Tests whether the references are returned as expected.
language/totto/table_to_text_utils_test.py