def it_repl_2d_vecs_1_col(): r = utils.repl_vec_over_cols(np.array([[1, 2, 3]]).T, 2) assert utils.np_array_same(r, expected)
def it_raises_on_non_vector(): with zest.raises(TypeError): utils.repl_vec_over_cols(np.array([[1, 2], [1, 2]]), 2) with zest.raises(TypeError): utils.repl_vec_over_cols(np.array([[[1, 2], [1, 2]]]), 2)
def it_repl_1d_vecs(): r = utils.repl_vec_over_cols(np.array([1, 2, 3]), 2) assert utils.np_array_same(r, expected)