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