+
−#!/usr/bin/env python3
+
−
+
−import sys
+
−sys.path.insert(0, '..')
+
−
+
−from tuikit.treeview import *
+
−import unittest
+
−
+
−
+
−class TestTreeView(unittest.TestCase):
+
−    def setUp(self):
+
−        """Build tree model
+
−
+
−        # root
+
−        # ├ a
+
−        # │ ├ c
+
−        # │ └ d
+
−        # │   ├ e
+
−        # │   │ └ g
+
−        # │   └ f
+
−        # │     └ h
+
−        # └ b
+
−
+
−        """
+
−        self.model = TreeModel()
+
−        self.model.add('/',  ['a', 'b'])
+
−        self.model.add(['a'], [TreeNode('c'), TreeNode('d')])
+
−        self.model.add((0,1), ['e', TreeNode('f')])
+
−        self.model.add([0,1,0], 'g')
+
−        self.model.add('/a/d/f', TreeNode('h'))
+
−
+
−    def test_treeiter(self):
+
−        """Iterate through the tree, test result."""
+
−        res = ''
+
−        for l, i, c, n in self.model:
+
−            res += str(l) + str(i) + str(c) + n.name
+
−
+
−        self.assertEqual(res, '112a212c222d312e411g322f411h122b')
+
−
+
−    def test_treemodel_find(self):
+
−        # good path
+
−        node = self.model.find('/a/d/f/h')
+
−        self.assertEqual(node.name, 'h')
+
−        self.assertEqual(node.path, '/a/d/f/h')
+
−        # bad path
+
−        self.assertRaises(ValueError, self.model.find, '/a/b/c')
+
−        self.assertRaises(IndexError, self.model.find, [0,1,3])
+
−
+
−if __name__ == '__main__':
+
−    unittest.main()
+
−