def test_get_axis(self): dat = DimArray(np.random.rand(5,10,3), dims=[Dim(range(5),name='one'), Dim(range(10),name='two'), Dim(range(3),name='three')],test='tst') self.assertEqual(dat.get_axis(0),0) self.assertEqual(dat.get_axis(1),1) self.assertEqual(dat.get_axis(2),2) self.assertEqual(dat.get_axis('one'),0) self.assertEqual(dat.get_axis('two'),1) self.assertEqual(dat.get_axis('three'),2)
def test_get_axis(self): dat = DimArray(np.random.rand(5,10,3), dims=[Dim(range(5),name='one'), Dim(range(10),name='two'), Dim(range(3),name='three')],test='tst') self.assertEqual(dat.get_axis(0),0) self.assertEqual(dat.get_axis(1),1) self.assertEqual(dat.get_axis(2),2) self.assertEqual(dat.get_axis('one'),0) self.assertEqual(dat.get_axis('two'),1) self.assertEqual(dat.get_axis('three'),2)