def test_issue121(self):
sh = (2, 1)
ureg = UnitRegistry()
z, v = 0, 2.
self.assertEqual(z + v * ureg.meter, v * ureg.meter)
self.assertEqual(z - v * ureg.meter, -v * ureg.meter)
self.assertEqual(v * ureg.meter + z, v * ureg.meter)
self.assertEqual(v * ureg.meter - z, v * ureg.meter)
self.assertEqual(sum([v * ureg.meter, v * ureg.meter]), 2 * v * ureg.meter)
z, v = np.zeros(sh), 2. * np.ones(sh)
self.assertQuantityEqual(z + v * ureg.meter, v * ureg.meter)
self.assertQuantityEqual(z - v * ureg.meter, -v * ureg.meter)
self.assertQuantityEqual(v * ureg.meter + z, v * ureg.meter)
self.assertQuantityEqual(v * ureg.meter - z, v * ureg.meter)
z, v = np.zeros((3, 1)), 2. * np.ones(sh)
for x, y in ((z, v),
(z, v * ureg.meter),
(v * ureg.meter, z)
):
try:
w = x + y
self.assertTrue(False, "ValueError not raised")
except ValueError:
pass
try:
w = x - y
self.assertTrue(False, "ValueError not raised")
except ValueError:
pass
def test_issue121b(self):
sh = (2, 1)
ureg = UnitRegistry()
z, v = 0, 2.0
self.assertEqual(z + v * ureg.meter, v * ureg.meter)
self.assertEqual(z - v * ureg.meter, -v * ureg.meter)
self.assertEqual(v * ureg.meter + z, v * ureg.meter)
self.assertEqual(v * ureg.meter - z, v * ureg.meter)
self.assertEqual(sum([v * ureg.meter, v * ureg.meter]),
2 * v * ureg.meter)
z, v = np.zeros(sh), 2.0 * np.ones(sh)
self.assertQuantityEqual(z + v * ureg.meter, v * ureg.meter)
self.assertQuantityEqual(z - v * ureg.meter, -v * ureg.meter)
self.assertQuantityEqual(v * ureg.meter + z, v * ureg.meter)
self.assertQuantityEqual(v * ureg.meter - z, v * ureg.meter)
z, v = np.zeros((3, 1)), 2.0 * np.ones(sh)
for x, y in ((z, v), (z, v * ureg.meter), (v * ureg.meter, z)):
with self.assertRaises(ValueError):
x + y
with self.assertRaises(ValueError):
x - y
def test_issue121(self):
sh = (2, 1)
ureg = UnitRegistry()
z, v = 0, 2.
self.assertEqual(z + v * ureg.meter, v * ureg.meter)
self.assertEqual(z - v * ureg.meter, -v * ureg.meter)
self.assertEqual(v * ureg.meter + z, v * ureg.meter)
self.assertEqual(v * ureg.meter - z, v * ureg.meter)
self.assertEqual(sum([v * ureg.meter, v * ureg.meter]),
2 * v * ureg.meter)
z, v = np.zeros(sh), 2. * np.ones(sh)
self.assertQuantityEqual(z + v * ureg.meter, v * ureg.meter)
self.assertQuantityEqual(z - v * ureg.meter, -v * ureg.meter)
self.assertQuantityEqual(v * ureg.meter + z, v * ureg.meter)
self.assertQuantityEqual(v * ureg.meter - z, v * ureg.meter)
z, v = np.zeros((3, 1)), 2. * np.ones(sh)
for x, y in ((z, v), (z, v * ureg.meter), (v * ureg.meter, z)):
try:
w = x + y
self.assertTrue(False, "ValueError not raised")
except ValueError:
pass
try:
w = x - y
self.assertTrue(False, "ValueError not raised")
except ValueError:
pass
def test_equal(self):
x = self.q.magnitude
u = self.Q_(np.ones(x.shape))
true = np.ones_like(x, dtype=np.bool_)
false = np.zeros_like(x, dtype=np.bool_)
self.assertQuantityEqual(u, u)
self.assertQuantityEqual(u == u, u.magnitude == u.magnitude)
self.assertQuantityEqual(u == 1, u.magnitude == 1)
v = self.Q_(np.zeros(x.shape), "m")
w = self.Q_(np.ones(x.shape), "m")
self.assertNDArrayEqual(v == 1, false)
self.assertNDArrayEqual(
self.Q_(np.zeros_like(x), "m") == self.Q_(np.zeros_like(x), "s"), false,
)
self.assertNDArrayEqual(v == v, true)
self.assertNDArrayEqual(v == w, false)
self.assertNDArrayEqual(v == w.to("mm"), false)
self.assertNDArrayEqual(u == v, false)
