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)
