| 123456789101112131415161718192021222324252627282930313233343536 |
- import unittest
- from node import Node, Leaf
- class TestNode(unittest.TestCase):
- def setUp(self):
- self.l0 = Leaf('leaf 1')
- self.l1 = Leaf('leaf 2')
- self.node = Node('node', self.l0, self.l1)
- def tearDown(self):
- pass
- def test_getitem(self):
- assert self.node[0] == self.l0
- def test_setitem(self):
- self.node[0] = Leaf('leaf 3')
- assert self.node[0] != self.l0
- def test_iter(self):
- s = ''
- for leaf in self.node:
- s += leaf.title()
- assert s == 'leaf 1leaf 2'
- def test_len(self):
- assert len(self.node) == 2
- def test_title(self):
- assert self.node.title() == 'node'
- assert self.l0.title() == 'leaf 1'
|
