def test_invalid_amount_of_operands(self):
        with pytest.raises(TypeError) as e:
            node = CombinationNode()
            node.append(PrimitiveNode(add))
            node.eval()

        assert "combination needs to have at least two operands" in str(e.value)
    def test_invalid_operator(self):
        with pytest.raises(TypeError) as e:
            node = CombinationNode()
            for i in range(3):
                node.append(PrimitiveNode(1))
            node.eval()

        assert "operator not found" in str(e.value)
예제 #3
0
 def test_eval_with_node(self):
     node = RootNode()
     combination = CombinationNode()
     combination.append(PrimitiveNode(add))
     combination.append(PrimitiveNode(1))
     combination.append(PrimitiveNode(2))
     node.append(combination)
     assert 3 == node.eval()
예제 #4
0
class RootNode(Node):
    def __init__(self):
        self._node = CombinationNode()

    def eval(self):
        return self._node.eval()

    def append(self, node):
        self._node = node
 def test_eval_without_nodes_should_return_none(self):
     node = CombinationNode()
     assert node.eval() is None
 def test_eval_two_operands(self):
     node = CombinationNode()
     node.append(PrimitiveNode(add))
     node.append(PrimitiveNode(1))
     node.append(PrimitiveNode(2))
     assert 3 == node.eval()
예제 #7
0
 def __init__(self):
     self._node = CombinationNode()