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'