Example #1
0
 def test_squeeze(self):
     self.assertQuantityEqual(np.squeeze(self.q), self.q)
     self.assertQuantityEqual(
         self.q.reshape([1, 4]).squeeze(), [1, 2, 3, 4] * self.ureg.m)
Example #2
0
 def test_squeeze(self):
     helpers.assert_quantity_equal(np.squeeze(self.q), self.q)
     helpers.assert_quantity_equal(
         self.q.reshape([1, 4]).squeeze(), [1, 2, 3, 4] * self.ureg.m
     )