expected_result = np.array([
        [[120, 120, 120], [6, 230, 230]],
        [[4, 250, 7], [204, 70, 3]]