예제 #1
0
def test_model_shape():

    shape = (10,11,12)
    ma = ModelArray()
    ma.model_params = np.zeros(shape+(4,))
    assert_equal(ma[..., 0].shape, shape[:-1])
    assert_equal(ma[0].shape, shape[1:])
    ma.shape = (-1)
    assert(ma.ndim == 1)
    ma.shape = shape[::-1]
    assert(ma.ndim == len(shape))
예제 #2
0
def test_model_shape():

    shape = (10, 11, 12)
    ma = ModelArray()
    ma.model_params = np.zeros(shape + (4, ))
    assert_equal(ma[..., 0].shape, shape[:-1])
    assert_equal(ma[0].shape, shape[1:])
    ma.shape = (-1)
    assert (ma.ndim == 1)
    ma.shape = shape[::-1]
    assert (ma.ndim == len(shape))