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