Ejemplos de DimArray.get_axis en Python

Ejemplo n.º 1

Archivo: test_dimarray.py Proyecto: maciekswat/ptsa

 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)

Ejemplo n.º 2

Archivo: test_dimarray.py Proyecto: isaac-ped/ptsa_new

 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)