def test_example_transposed(self): A = np.array([ [1,1,3,3,3,3], [2,2,4,3,3,3], [3,3,5,3,4,4], [4,4,6,3,5,5]]) A_rl, n = rlencode(A, 1) assert np.array_equal(A_rl.shape, np.array([4, 4])) assert n.size == 4 assert np.array_equal(A_rl, np.array([ [1,3,3,3], [2,4,3,3], [3,5,3,4], [4,6,3,5]])) assert np.array_equal(n, np.array([2,1,1,2]))
def test_example(self): A = np.array([ [1,2,3,4], [1,2,3,4], [3,4,5,6], [3,3,3,3], [3,3,4,5], [3,3,4,5]]) A_rl, n = rlencode(A, 0) assert np.array_equal(A_rl.shape, np.array([4, 4])) assert n.size == 4 assert np.array_equal(A_rl, np.array([ [1,2,3,4], [3,4,5,6], [3,3,3,3], [3,3,4,5]])) assert np.array_equal(n, np.array([2,1,1,2]))
def test_1d_array(self): A = np.array([1,1,1,1,4,4,5]) A_rl, n = rlencode(A) assert np.array_equal(A_rl, np.array([1,4,5])) assert np.array_equal(n, np.array([4,2,1]))