1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
# (C) Copyright 2004-2023 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!
"""
Test cases for the TreeNode object.
"""
import unittest
from traits.api import HasStrictTraits, List, Str, This
from traits.testing.api import UnittestTools
from traitsui.api import TreeNode
from traitsui.tests._tools import BaseTestMixin
class DummyModel(HasStrictTraits):
"""Dummy model with children."""
name = Str()
children = List(This)
class TestTreeNode(BaseTestMixin, UnittestTools, unittest.TestCase):
def setUp(self):
BaseTestMixin.setUp(self)
def tearDown(self):
BaseTestMixin.tearDown(self)
def test_insert_child(self):
# Regression test for #559
model = DummyModel(
name="Parent",
children=[DummyModel(name="Child0"), DummyModel(name="Child2")],
)
node = TreeNode(children="children", node_for=[DummyModel])
node.insert_child(model, 1, DummyModel(name="Child1"))
# Assert
self.assertEqual(len(model.children), 3)
for i in range(3):
self.assertEqual(model.children[i].name, "Child{}".format(i))
if __name__ == "__main__":
unittest.run()
|
