openstack
/
deb-python-pint
Code Issues Proposed changes

add __contains__ to UnitsContainer

This commit is contained in:
Robert DeNeefe 2017-05-22 21:52:22 -05:00
parent 4c5b373548
commit ed687f8405
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pint/testsuite/test_quantity.py
View File

@ -595,6 +595,18 @@ class TestDimensions(QuantityTestCase):
self.assertFalse((self.Q_(42, 'meter') / self.Q_(1, 'second')).dimensionless)
self.assertTrue((self.Q_(42, 'meter') / self.Q_(1, 'inch')).dimensionless)
def test_inclusion(self):
dim = self.Q_(42, 'meter').dimensionality
self.assertTrue('[length]' in dim)
self.assertFalse('[time]' in dim)
dim = (self.Q_(42, 'meter') / self.Q_(11, 'second')).dimensionality
self.assertTrue('[length]' in dim)
self.assertTrue('[time]' in dim)
dim = self.Q_(20.785, 'J/(mol)').dimensionality
for dimension in ('[length]', '[mass]', '[substance]', '[time]'):
self.assertTrue(dimension in dim)
self.assertFalse('[angle]' in dim)
class TestQuantityWithDefaultRegistry(TestDimensions):

3
pint/util.py
View File

@ -297,6 +297,9 @@ class UnitsContainer(Mapping):
def __getitem__(self, key):
return self._d[key]
def __contains__(self, key):
return key in self._d
def __hash__(self):
return self._hash