def testUpside_down_flip(self):
squarelotron_4x4 = squarelotron.make_4x4_squarelotron()
squarelotron_5x5 = squarelotron.make_5x5_squarelotron()
self.assertEquals(squarelotron.upside_down_flip(squarelotron_4x4, 0),
[[13, 14, 15, 16],
[9, 6, 7, 12],
[5, 10, 11, 8],
[1, 2, 3, 4]])
self.assertEquals(
[[1, 2, 3, 4],
[5, 10, 11, 8],
[9, 6, 7, 12],
[13, 14, 15, 16]],
squarelotron.upside_down_flip(squarelotron_4x4, 1))
self.assertEquals(squarelotron.upside_down_flip(squarelotron_5x5, 0),
[[21, 22, 23, 24, 25],
[16, 7, 8, 9, 20],
[11, 12, 13, 14, 15],
[6, 17, 18, 19, 10],
[1, 2, 3, 4, 5]])
self.assertEquals(
[[1, 2, 3, 4, 5],
[6, 17, 18, 19, 10],
[11, 12, 13, 14, 15],
[16, 7, 8, 9, 20],
[21, 22, 23, 24, 25]],
squarelotron.upside_down_flip(squarelotron_5x5, 1))
def testLeft_right_flip(self):
squarelotron_4x4 = squarelotron.make_4x4_squarelotron()
squarelotron_5x5 = squarelotron.make_5x5_squarelotron()
self.assertEquals(
[[4, 3, 2, 1],
[8, 6, 7, 5],
[12, 10, 11, 9],
[16, 15, 14, 13]],
squarelotron.left_right_flip(squarelotron_4x4, 0))
self.assertEquals(
[[1, 2, 3, 4],
[5, 7, 6, 8],
[9, 11, 10, 12],
[13, 14, 15, 16]],
squarelotron.left_right_flip(squarelotron_4x4, 1))
self.assertEquals(
[[5, 4, 3, 2, 1],
[10, 7, 8, 9, 6],
[15, 12, 13, 14, 11],
[20, 17, 18, 19, 16],
[25, 24, 23, 22, 21]],
squarelotron.left_right_flip(squarelotron_5x5, 0))
self.assertEquals(
[[1, 2, 3, 4, 5],
[6, 9, 8, 7, 10],
[11, 14, 13, 12, 15],
[16, 19, 18, 17, 20],
[21, 22, 23, 24, 25]],
squarelotron.left_right_flip(squarelotron_5x5, 1))
def testMake_5x5_squarelotron(self):
self.assertEquals([[1, 2, 3, 4, 5],
[6, 7, 8, 9, 10],
[11, 12, 13, 14, 15],
[16, 17, 18, 19, 20],
[21, 22, 23, 24, 25]],
squarelotron.make_5x5_squarelotron())
def testMain_diagonal_flip(self):
squarelotron_4x4 = squarelotron.make_4x4_squarelotron()
squarelotron_5x5 = squarelotron.make_5x5_squarelotron()
self.assertEquals(
[[1, 2, 3, 4],
[5, 6, 10, 8],
[9, 7, 11, 12],
[13, 14, 15, 16]],
squarelotron.main_diagonal_flip(squarelotron_4x4, 1))
self.assertEquals(
[[1, 5, 9, 13],
[2, 6, 7, 14],
[3, 10, 11, 15],
[4, 8, 12, 16]],
squarelotron.main_diagonal_flip(squarelotron_4x4, 0))
self.assertEquals(
[[1, 6, 11, 16, 21],
[2, 7, 8, 9, 22],
[3, 12, 13, 14, 23],
[4, 17, 18, 19, 24],
[5, 10, 15, 20, 25]],
squarelotron.main_diagonal_flip(squarelotron_5x5, 0))
self.assertEquals(
[[ 1, 2, 3, 4, 5],
[ 6, 7, 12, 17, 10],
[11, 8, 13, 18, 15],
[16, 9, 14, 19, 20],
[21, 22, 23, 24, 25]],
squarelotron.main_diagonal_flip(squarelotron_5x5, 1))
def testInverse_diagonal_flip(self):
squarelotron_4x4 = squarelotron.make_4x4_squarelotron()
squarelotron_5x5 = squarelotron.make_5x5_squarelotron()
self.assertEquals(
[[1, 2, 3, 4],
[5, 11, 7, 8],
[9, 10, 6, 12],
[13, 14, 15, 16]],
squarelotron.inverse_diagonal_flip(squarelotron_4x4, 1))
self.assertEquals(
[[16, 12, 8, 4],
[15, 6, 7, 3],
[14, 10, 11, 2],
[13, 9, 5, 1]],
squarelotron.inverse_diagonal_flip(squarelotron_4x4, 0))
self.assertEquals(
[[25, 20, 15, 10, 5],
[24, 7, 8, 9, 4],
[23, 12, 13, 14, 3],
[22, 17, 18, 19, 2],
[21, 16, 11, 6, 1]],
squarelotron.inverse_diagonal_flip(squarelotron_5x5, 0))
self.assertEquals(
[[ 1, 2, 3, 4, 5],
[ 6, 19, 14, 9, 10],
[11, 18, 13, 8, 15],
[16, 17, 12, 7, 20],
[21, 22, 23, 24, 25]],
squarelotron.inverse_diagonal_flip(squarelotron_5x5, 1))
def testGet_column(self):
squarelotron_4x4 = squarelotron.make_4x4_squarelotron()
squarelotron_5x5 = squarelotron.make_5x5_squarelotron()
self.assertEquals([1,6,11,16,21],
squarelotron.get_column(squarelotron_5x5, 0))
self.assertEquals([4, 8, 12, 16],
squarelotron.get_column(squarelotron_4x4, 3))
def testMake_list(self):
self.assertEquals(
range(1,17),
squarelotron.make_list(squarelotron.make_4x4_squarelotron()))
self.assertEquals(
range(1,26),
squarelotron.make_list(squarelotron.make_5x5_squarelotron()))
