class TestSpanFinding(object):
def setup(self):
self.grid = Grid(two_by_three)
def test_right_nowrap(self):
found = list(self.grid._right_span((1, 0), 3, False))
assert found == [(1, 0), (1, 1), (1, 2)]
def test_right_wrap(self):
found = list(self.grid._right_span((0, 2), 3, True))
assert found == [(0, 2), (0, 0), (0, 1)]
def test_right_nowrap_nospace(self):
"""Should return None b/c there's not enough space for a span."""
found = self.grid._right_span((1, 1), 3, False)
assert found == None
def test_left(self):
found = self.grid._left_span((0, 2), 3, False)
print found
assert found == [(0, 2), (0, 1), (0, 0)]
def test_left_wrap(self):
found = self.grid._left_span((1, 0), 3, True)
print found
assert found == [(1, 0), (1, 2), (1, 1)]
def test_up(self):
found = self.grid._up_span((1, 0), 2, False)
print found
assert found == [(1, 0), (0, 0)]
def test_up_wrap(self):
found = self.grid._up_span((0, 0), 2, True)
assert found == [(0, 0), (1, 0)]
def test_up_too_long(self):
assert self.grid._up_span((0, 0), 2, False) is None
def test_down(self):
found = self.grid._down_span((0, 0), 2, False)
assert found == [(0, 0), (1, 0)]
def test_down_wrap(self):
found = self.grid._down_span((1, 0), 2, True)
assert found == [(1, 0), (0, 0)]
def test_down_too_long(self):
assert self.grid._down_span((0, 0), 3, False) is None
def test_down_too_long_with_wrap(self):
assert self.grid._down_span((0, 0), 3, True) is None
def test_down_over_end(self):
assert self.grid._down_span((1, 0), 2, False) is None
def test_num_spans_no_wrap(self):
assert len(list(self.grid.spans((0, 0), 2, False))) == 3
def test_num_spans_wrap(self):
assert len(list(self.grid.spans((0, 0), 2, True))) == 8
def test_right_wrap(self):
"""Should find a span that wraps around the right side."""
found = list(self.grid._right_span((1, 1), 3, True))
print found
assert found == [(1, 1), (1, 2), (1, 0)]
def test_right_wrap_with_no_space(self):
"""Should return None because a square can't be used twice."""
assert self.grid._right_span((1, 1), 4, True) is None
def test_str(self):
assert str(self.grid) == 'CAT\nNTA'
