| 1234567891011121314151617181920212223242526272829303132333435363738394041424344 |
- 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'
- def test____str__(self):
- l1 = Leaf(1)
- self.assertEqual(str(l1), '1')
- l1 = -l1
- self.assertEqual(str(l1), '-1')
- l1 = -l1
- self.assertEqual(str(l1), '--1')
|
