def test_convert_to_consistent_units_with_dimensionality_error(self):
with pytest.raises(DimensionalityError):
convert_to_consistent_units(
self.Q_(32.0, "degF"),
pre_calc_units=self.ureg.meter,
)
with pytest.raises(DimensionalityError):
convert_to_consistent_units(
np.arange(4),
pre_calc_units=self.ureg.meter,
)
def test_convert_to_consistent_units_without_pre_calc_units(self):
args, kwargs = convert_to_consistent_units(
(self.Q_(0), self.Q_(10, "degC")),
[1, 2, 3, 5, 7] * self.ureg.m,
pre_calc_units=None,
)
assert args[0][0] == 0
assert args[0][1] == 10
self.assertNDArrayEqual(args[1], np.array([1, 2, 3, 5, 7]))
assert kwargs == {}
def test_convert_to_consistent_units_with_pre_calc_units(self):
args, kwargs = convert_to_consistent_units(
self.Q_(50, "cm"),
np.arange(4).reshape(2, 2) * self.ureg.m,
[0.042] * self.ureg.km,
(self.Q_(0, "m"), self.Q_(1, "dam")),
a=6378 * self.ureg.km,
pre_calc_units=self.ureg.meter,
)
assert args[0] == 0.5
self.assertNDArrayEqual(args[1], np.array([[0, 1], [2, 3]]))
self.assertNDArrayEqual(args[2], np.array([42]))
assert args[3][0] == 0
assert args[3][1] == 10
assert kwargs["a"] == 6.378e6
def test_convert_to_consistent_units_with_dimensionless(self):
args, kwargs = convert_to_consistent_units(np.arange(2),
pre_calc_units=self.ureg.g /
self.ureg.kg)
self.assertNDArrayEqual(args[0], np.array([0, 1000]))
assert kwargs == {}