#!/usr/bin/env python3

from tuikit.treeview import *
import unittest

class TestTreeView(unittest.TestCase):
    def test_treemodel(self):
        '''Build tree model, iterate through the tree, test result.'''
        # build tree model
        # root
        # ├ a
        # │ ├ c
        # │ └ d
        # │   ├ e
        # │   │ └ g
        # │   └ f
        # │     └ h
        # └ b
        model = TreeModel()
        model.add('/',  ['a', 'b'])
        model.add('/a', ['c', 'd'])
        model.add((0,1), ['e', 'f'])
        model.add('/0/1/0', 'g')
        model.add('/a/d/f', 'h')
        
        res = ''
        for l, i, c, n in model:
            res += str(l) + str(i) + str(c) + n.name
            
        self.assertEqual(res, '112a212c222d312e411g322f411h122b')


if __name__ == '__main__':
    unittest.main()