def testCellsForColumnInvalidColumn(self): connections = Connections([64, 64], 4) try: connections.cellsForColumn(4095) except IndexError: self.fail("IndexError raised unexpectedly") args = [4096] self.assertRaises(IndexError, connections.cellsForColumn, *args) args = [-1] self.assertRaises(IndexError, connections.cellsForColumn, *args)
def testCellsForColumn2D(self): connections = Connections([64, 64], 4) expectedCells = set([256, 257, 258, 259]) self.assertEqual(connections.cellsForColumn(64), expectedCells)
def testCellsForColumn1D(self): connections = Connections([2048], 5) expectedCells = set([5, 6, 7, 8, 9]) self.assertEqual(connections.cellsForColumn(1), expectedCells)