

def test_expand_dims_vector(self):
    m = np.random.rand(3)
    x = mu.expand_dims(np.copy(m), 1)
    assert_equals(x.ndim, 1)
    assert_equals(x.shape, (3,))
    x = mu.expand_dims(np.copy(m), 2)
    assert_equals(x.ndim, 2)
    assert_equals(x.shape, (1, 3))
    x = mu.expand_dims(np.copy(m), 2)
    assert_equals(x.ndim, 2)
    assert_equals(x.shape, (1, 3))

dataset/ETHPy150Open kashefy/nideep/nideep/blobs/test_mat_utils.py TestExpandDims.test_expand_dims_vector/original