Skip to content

Commit

Permalink
add tests fro tree
Browse files Browse the repository at this point in the history
  • Loading branch information
davidbailey committed Feb 3, 2019
1 parent 7aa4b62 commit dde6ed0
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions rail/tests/test_tree.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
import unittest

from rail import Tree

class TestTree(unittest.TestCase):
def setUp(self):
self.tree = Tree(name='test tree')
self.tree.addChild('test child')

def test_tree(self):
self.assertEqual(self.tree.name, 'test tree')
self.assertEqual(self.tree.parent, None)
self.assertEqual(self.tree['test child'].name, 'test child')
self.assertEqual(self.tree['test child'].parent, self.tree)

def test_path(self):
self.assertEqual(self.tree['test child'].path(), '/test tree/test child')

def test_to_dict_list(self):
self.assertEqual(self.tree.to_dict_list(), {'name': 'test tree', 'children': [{'name': 'test child', 'children': []}]})

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

0 comments on commit dde6ed0

Please sign in to comment.