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], ]])
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], ] ] )