class TestCompatibleUnits(QuantityTestCase):
FORCE_NDARRAY = False
def setUp(self):
super(TestCompatibleUnits, self).setUp()
self.ureg = UnitRegistry(force_ndarray=self.FORCE_NDARRAY)
self.Q_ = self.ureg.Quantity
self.U_ = self.ureg.Unit
def _test(self, input_units):
gd = self.ureg.get_dimensionality
dim = gd(input_units)
equiv = self.ureg.get_compatible_units(input_units)
for eq in equiv:
self.assertEqual(gd(eq), dim)
self.assertEqual(equiv, self.ureg.get_compatible_units(dim))
def _test2(self, units1, units2):
equiv1 = self.ureg.get_compatible_units(units1)
equiv2 = self.ureg.get_compatible_units(units2)
self.assertEqual(equiv1, equiv2)
def test_many(self):
self._test(self.ureg.meter)
self._test(self.ureg.seconds)
self._test(self.ureg.newton)
self._test(self.ureg.kelvin)
def test_context_sp(self):
gd = self.ureg.get_dimensionality
# length, frequency, energy
valid = [
gd(self.ureg.meter),
gd(self.ureg.hertz),
gd(self.ureg.joule), 1 / gd(self.ureg.meter)
]
with self.ureg.context('sp'):
equiv = self.ureg.get_compatible_units(self.ureg.meter)
result = set()
for eq in equiv:
dim = gd(eq)
result.add(dim)
self.assertIn(dim, valid)
self.assertEqual(len(result), len(valid))
def test_get_base_units(self):
ureg = UnitRegistry()
self.assertEqual(ureg.get_base_units(''), (1, ureg.Unit('')))
self.assertEqual(ureg.get_base_units('meter'),
ureg.get_base_units(ParserHelper(meter=1)))
def test_get_compatible_units(self):
ureg = UnitRegistry()
self.assertEqual(ureg.get_compatible_units(''), frozenset())
self.assertEqual(ureg.get_compatible_units('meter'),
ureg.get_compatible_units(ParserHelper(meter=1)))
class TestCompatibleUnits(QuantityTestCase):
FORCE_NDARRAY = False
def setUp(self):
super(TestCompatibleUnits, self).setUp()
self.ureg = UnitRegistry(force_ndarray=self.FORCE_NDARRAY)
self.Q_ = self.ureg.Quantity
self.U_ = self.ureg.Unit
def _test(self, input_units):
gd = self.ureg.get_dimensionality
dim = gd(input_units)
equiv = self.ureg.get_compatible_units(input_units)
for eq in equiv:
self.assertEqual(gd(eq), dim)
self.assertEqual(equiv, self.ureg.get_compatible_units(dim))
def _test2(self, units1, units2):
equiv1 = self.ureg.get_compatible_units(units1)
equiv2 = self.ureg.get_compatible_units(units2)
self.assertEqual(equiv1, equiv2)
def test_many(self):
self._test(self.ureg.meter)
self._test(self.ureg.seconds)
self._test(self.ureg.newton)
self._test(self.ureg.kelvin)
def test_context_sp(self):
gd = self.ureg.get_dimensionality
# length, frequency, energy
valid = [gd(self.ureg.meter), gd(self.ureg.hertz),
gd(self.ureg.joule), 1/gd(self.ureg.meter)]
with self.ureg.context('sp'):
equiv = self.ureg.get_compatible_units(self.ureg.meter)
result = set()
for eq in equiv:
dim = gd(eq)
result.add(dim)
self.assertIn(dim, valid)
self.assertEqual(len(result), len(valid))
def test_get_base_units(self):
ureg = UnitRegistry()
self.assertEqual(ureg.get_base_units(''), (1, ureg.Unit('')))
self.assertEqual(ureg.get_base_units('meter'), ureg.get_base_units(ParserHelper(meter=1)))
def test_get_compatible_units(self):
ureg = UnitRegistry()
self.assertEqual(ureg.get_compatible_units(''), frozenset())
self.assertEqual(ureg.get_compatible_units('meter'),
ureg.get_compatible_units(ParserHelper(meter=1)))
