Python expand_dims Examples

Example #1

 def test_expand_dims(self):
     self.assertQuantityEqual(np.expand_dims(self.q, 0),
                              np.array([[[1, 2], [3, 4]]]) * self.ureg.m)

Example #2

File: test_numpy.py Project: wiki01/pint

 def test_expand_dims(self):
     helpers.assert_quantity_equal(
         np.expand_dims(self.q, 0), np.array([[[1, 2], [3, 4]]]) * self.ureg.m
     )