Example #1
0
 def test_rollaxis(self):
     self.assertQuantityEqual(np.rollaxis(self.q, 1),
                              np.array([[1, 2], [3, 4]]).T * self.ureg.m)
Example #2
0
 def test_rollaxis(self):
     helpers.assert_quantity_equal(
         np.rollaxis(self.q, 1), np.array([[1, 2], [3, 4]]).T * self.ureg.m
     )