| 1234567891011121314151617 |
- import unittest
- from src.node import ExpressionNode as N, ExpressionLeaf as L
- from src.rules.utils import nary_node
- class TestRules(unittest.TestCase):
- def test_nary_node_binary(self):
- l0, l1 = L(1), L(2)
- plus = N('+', l0, l1)
- self.assertEqual(nary_node('+', [l0, l1]), plus)
- def test_nary_node_ternary(self):
- l0, l1, l2 = L(1), L(2), L(3)
- plus = N('+', N('+', l0, l1), l2)
- self.assertEqual(nary_node('+', [l0, l1, l2]), plus)
|
