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))
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))