def test_add_dim(self):
# make ndarray an Dimaray with identical data
arr = np.random.rand(5)
dat = DimArray(arr,dims=[Dim(range(5),name='one')])
# make new dim to add
d = Dim(range(10),name='replicate')
# add it to the dat
ndat = dat.add_dim(d)
# test that it worked
# verify shape
self.assertEquals(len(ndat.shape),len(dat.shape)+1)
self.assertEquals(ndat.shape[0],10)
self.assertEquals(ndat.shape[1],5)
# verify contents (a couple random spots)
assert_array_equal(ndat[4],dat)
assert_array_equal(ndat[7],dat)
assert_array_equal(ndat.dims[0],d)
assert_array_equal(ndat.dims[1],dat.dims[0])
def test_add_dim(self):
# make ndarray an Dimaray with identical data
arr = np.random.rand(5)
dat = DimArray(arr,dims=[Dim(range(5),name='one')])
# make new dim to add
d = Dim(range(10),name='replicate')
# add it to the dat
ndat = dat.add_dim(d)
# test that it worked
# verify shape
self.assertEquals(len(ndat.shape),len(dat.shape)+1)
self.assertEquals(ndat.shape[0],10)
self.assertEquals(ndat.shape[1],5)
# verify contents (a couple random spots)
assert_array_equal(ndat[4],dat)
assert_array_equal(ndat[7],dat)
assert_array_equal(ndat.dims[0],d)
assert_array_equal(ndat.dims[1],dat.dims[0])
