Merge pull request #207 from jonls/boolean-length

boolean: Allow getting length of And/Or terms

Author: jonls

psamm/expression/boolean.py

@@ -78,6 +78,9 @@ def __iter__(self):
     def __hash__(self):
         return hash(self._terms)
 
+    def __len__(self):
+        return len(self._terms)
+
     def __eq__(self, other):
         return (isinstance(other, self.__class__) and
                 self._terms == other._terms)

psamm/tests/test_expression_boolean.py

@@ -62,6 +62,16 @@ def test_variable_hash(self):
         self.assertEqual(hash(Variable('xyz')), hash(Variable('xyz')))
 
 
+class TestTerm(unittest.TestCase):
+    def test_length_of_and(self):
+        term = And(Variable('b1'), Variable('b2'))
+        self.assertEqual(len(term), 2)
+
+    def test_length_of_or(self):
+        term = Or(Variable('b1'), And(Variable('b2'), Variable('b3')))
+        self.assertEqual(len(term), 2)
+
+
 class TestExpression(unittest.TestCase):
     def test_expression_substitute_existing(self):
         e = Expression('b1')