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 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()))
def testMake_4x4_squarelotron(self): self.assertEquals([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]], squarelotron.make_4x4_squarelotron())