class TestGridOperations(object):
def setup(self):
self.grid = Grid(two_by_three)
def test_num_row_and_cols(self):
assert self.grid.num_rows == 2
assert self.grid.num_cols == 3
def test_getitem(self):
assert self.grid[0, 1] == 'A'
def test_letters_at_indices(self):
found = list(self.grid.letters_at_indices([(0, 1), (0, 2)]))
assert found == ['A', 'T']
def test_word_at_indices(self):
found = self.grid.word_at_indices([(0, 1), (0, 2)])
print found
assert found == 'AT'
def test_positions_that_have_letter(self):
found = list(self.grid.positions_that_have_letter('T'))
print found
assert found == [(0, 2), (1, 1)]
def test_iter(self):
found = list(iter(self.grid))
assert found == ['C', 'A', 'T', 'N', 'T', 'A']
def test_index_letter_pairs(self):
found = list(self.grid.index_letter_pairs)
print found
assert found == [((0, 0), 'C'), ((0, 1), 'A'), ((0, 2), 'T'),
((1, 0), 'N'), ((1, 1), 'T'), ((1, 2), 'A')]
