コード例 #1
0
 def test_atf(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, -1, -1, -1, -1],
         [-1, 1, -1, -1, -1],
     ])
     assert_array_equal(LTFArray.transform_atf(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, 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
0
 def test_atf(self):
     """This method tests the atf 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, -1, -1, -1, -1],
         [-1, 1, -1, -1, -1],
     ])
     assert_array_equal(
         LTFArray.transform_atf(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, 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],
             ]
         ]
     )