class SpiderTest(unittest.TestCase):
def setUp(self):
self.grid = Grid(10)
self.spider = Spider(self.grid)
node_cell = self.spider.nodes[0]
node_cell.remove_node()
self.spider.nodes.clear()
self.spider.pos = None
'''
0 1 2 3 4 5 6 7 8 9
0 - - - - - - - - - -
1 - - - - - - - - - -
2 - - - - - - - - - -
3 - - - - - - - - - -
4 - - - - - - - - - -
5 - - - - - - - - - -
6 - - - - - - - - - -
7 - - - - - - - - - -
8 - - - - - - - - - -
9 - - - - - - - - - -
'''
def test_choose_node_cell(self):
self.spider.pos = Vector(0, 0)
valid_cells = (Vector(0, 2), Vector(0, 3), Vector(0, 4))
for _ in range(20):
node_cell = self.spider.choose_node_cell("RIGHT", Vector(0, 4))
self.assertIn(node_cell.pos, valid_cells)
