コード例 #1
0
 def test_shift(self):
     test_array = array([
         [-1, 1, 1, -1, -1],
         [-1, -1, -1, -1, -1],
         [1, 1, -1, -1, -1],
         [1, 1, -1, -1, -1],
         [1, 2, 3, 4, 5],
     ])
     assert_array_equal(LTFArray.transform_shift(test_array, k=3), [[
         [-1, 1, 1, -1, -1],
         [1, 1, -1, -1, -1],
         [1, -1, -1, -1, 1],
     ], [
         [-1, -1, -1, -1, -1],
         [-1, -1, -1, -1, -1],
         [-1, -1, -1, -1, -1],
     ], [
         [1, 1, -1, -1, -1],
         [1, -1, -1, -1, 1],
         [-1, -1, -1, 1, 1],
     ], [
         [1, 1, -1, -1, -1],
         [1, -1, -1, -1, 1],
         [-1, -1, -1, 1, 1],
     ], [
         [1, 2, 3, 4, 5],
         [2, 3, 4, 5, 1],
         [3, 4, 5, 1, 2],
     ]])
コード例 #2
0
 def test_shift(self):
     """This method tests the shift transformation with predefined input and output."""
     test_array = array([
         [-1, 1, 1, -1, -1],
         [-1, -1, -1, -1, -1],
         [1, 1, -1, -1, -1],
         [1, 1, -1, -1, -1],
         [1, 2, 3, 4, 5],
     ],
                        dtype=tools.BIT_TYPE)
     assert_array_equal(LTFArray.transform_shift(test_array, k=3), [[
         [-1, 1, 1, -1, -1],
         [1, 1, -1, -1, -1],
         [1, -1, -1, -1, 1],
     ], [
         [-1, -1, -1, -1, -1],
         [-1, -1, -1, -1, -1],
         [-1, -1, -1, -1, -1],
     ], [
         [1, 1, -1, -1, -1],
         [1, -1, -1, -1, 1],
         [-1, -1, -1, 1, 1],
     ], [
         [1, 1, -1, -1, -1],
         [1, -1, -1, -1, 1],
         [-1, -1, -1, 1, 1],
     ], [
         [1, 2, 3, 4, 5],
         [2, 3, 4, 5, 1],
         [3, 4, 5, 1, 2],
     ]])