Show More
@@ -1,194 +1,214 b'' | |||||
1 | #!/usr/bin/env python |
|
1 | #!/usr/bin/env python | |
2 | # encoding: utf-8 |
|
2 | # encoding: utf-8 | |
3 | """ |
|
3 | """ | |
4 | Tests for IPython.core.component |
|
4 | Tests for IPython.core.component | |
5 |
|
5 | |||
6 | Authors: |
|
6 | Authors: | |
7 |
|
7 | |||
8 | * Brian Granger |
|
8 | * Brian Granger | |
9 | * Fernando Perez (design help) |
|
9 | * Fernando Perez (design help) | |
10 | """ |
|
10 | """ | |
11 |
|
11 | |||
12 | #----------------------------------------------------------------------------- |
|
12 | #----------------------------------------------------------------------------- | |
13 | # Copyright (C) 2008-2009 The IPython Development Team |
|
13 | # Copyright (C) 2008-2009 The IPython Development Team | |
14 | # |
|
14 | # | |
15 | # Distributed under the terms of the BSD License. The full license is in |
|
15 | # Distributed under the terms of the BSD License. The full license is in | |
16 | # the file COPYING, distributed as part of this software. |
|
16 | # the file COPYING, distributed as part of this software. | |
17 | #----------------------------------------------------------------------------- |
|
17 | #----------------------------------------------------------------------------- | |
18 |
|
18 | |||
19 | #----------------------------------------------------------------------------- |
|
19 | #----------------------------------------------------------------------------- | |
20 | # Imports |
|
20 | # Imports | |
21 | #----------------------------------------------------------------------------- |
|
21 | #----------------------------------------------------------------------------- | |
22 |
|
22 | |||
23 | from unittest import TestCase |
|
23 | from unittest import TestCase | |
24 |
|
24 | |||
25 | from IPython.core.component import Component, ComponentError |
|
25 | from IPython.core.component import Component, ComponentError | |
26 | from IPython.utils.traitlets import ( |
|
26 | from IPython.utils.traitlets import ( | |
27 | TraitletError, Int, Float, Str |
|
27 | TraitletError, Int, Float, Str | |
28 | ) |
|
28 | ) | |
29 | from IPython.config.loader import Config |
|
29 | from IPython.config.loader import Config | |
30 |
|
30 | |||
31 |
|
31 | |||
32 | #----------------------------------------------------------------------------- |
|
32 | #----------------------------------------------------------------------------- | |
33 | # Test cases |
|
33 | # Test cases | |
34 | #----------------------------------------------------------------------------- |
|
34 | #----------------------------------------------------------------------------- | |
35 |
|
35 | |||
36 |
|
36 | |||
37 | class TestComponentMeta(TestCase): |
|
37 | class TestComponentMeta(TestCase): | |
38 |
|
38 | |||
39 | def test_get_instances(self): |
|
39 | def test_get_instances(self): | |
40 | class BaseComponent(Component): |
|
40 | class BaseComponent(Component): | |
41 | pass |
|
41 | pass | |
42 | c1 = BaseComponent(None) |
|
42 | c1 = BaseComponent(None) | |
43 | c2 = BaseComponent(c1) |
|
43 | c2 = BaseComponent(c1) | |
44 | self.assertEquals(BaseComponent.get_instances(),[c1,c2]) |
|
44 | self.assertEquals(BaseComponent.get_instances(),[c1,c2]) | |
45 |
|
45 | |||
46 | def test_get_instances_subclass(self): |
|
46 | def test_get_instances_subclass(self): | |
47 | class MyComponent(Component): |
|
47 | class MyComponent(Component): | |
48 | pass |
|
48 | pass | |
49 | class MyOtherComponent(MyComponent): |
|
49 | class MyOtherComponent(MyComponent): | |
50 | pass |
|
50 | pass | |
51 | c1 = MyComponent(None) |
|
51 | c1 = MyComponent(None) | |
52 | c2 = MyOtherComponent(c1) |
|
52 | c2 = MyOtherComponent(c1) | |
53 | c3 = MyOtherComponent(c2) |
|
53 | c3 = MyOtherComponent(c2) | |
54 | self.assertEquals(MyComponent.get_instances(), [c1, c2, c3]) |
|
54 | self.assertEquals(MyComponent.get_instances(), [c1, c2, c3]) | |
55 | self.assertEquals(MyOtherComponent.get_instances(), [c2, c3]) |
|
55 | self.assertEquals(MyOtherComponent.get_instances(), [c2, c3]) | |
56 |
|
56 | |||
57 | def test_get_instances_root(self): |
|
57 | def test_get_instances_root(self): | |
58 | class MyComponent(Component): |
|
58 | class MyComponent(Component): | |
59 | pass |
|
59 | pass | |
60 | class MyOtherComponent(MyComponent): |
|
60 | class MyOtherComponent(MyComponent): | |
61 | pass |
|
61 | pass | |
62 | c1 = MyComponent(None) |
|
62 | c1 = MyComponent(None) | |
63 | c2 = MyOtherComponent(c1) |
|
63 | c2 = MyOtherComponent(c1) | |
64 | c3 = MyOtherComponent(c2) |
|
64 | c3 = MyOtherComponent(c2) | |
65 | c4 = MyComponent(None) |
|
65 | c4 = MyComponent(None) | |
66 | c5 = MyComponent(c4) |
|
66 | c5 = MyComponent(c4) | |
67 | self.assertEquals(MyComponent.get_instances(root=c1), [c1, c2, c3]) |
|
67 | self.assertEquals(MyComponent.get_instances(root=c1), [c1, c2, c3]) | |
68 | self.assertEquals(MyComponent.get_instances(root=c4), [c4, c5]) |
|
68 | self.assertEquals(MyComponent.get_instances(root=c4), [c4, c5]) | |
69 |
|
69 | |||
70 |
|
70 | |||
71 | class TestComponent(TestCase): |
|
71 | class TestComponent(TestCase): | |
72 |
|
72 | |||
73 | def test_parent_child(self): |
|
73 | def test_parent_child(self): | |
74 | c1 = Component(None) |
|
74 | c1 = Component(None) | |
75 | c2 = Component(c1) |
|
75 | c2 = Component(c1) | |
76 | c3 = Component(c1) |
|
76 | c3 = Component(c1) | |
77 | c4 = Component(c3) |
|
77 | c4 = Component(c3) | |
78 | self.assertEquals(c1.parent, None) |
|
78 | self.assertEquals(c1.parent, None) | |
79 | self.assertEquals(c2.parent, c1) |
|
79 | self.assertEquals(c2.parent, c1) | |
80 | self.assertEquals(c3.parent, c1) |
|
80 | self.assertEquals(c3.parent, c1) | |
81 | self.assertEquals(c4.parent, c3) |
|
81 | self.assertEquals(c4.parent, c3) | |
82 | self.assertEquals(c1.children, [c2, c3]) |
|
82 | self.assertEquals(c1.children, [c2, c3]) | |
83 | self.assertEquals(c2.children, []) |
|
83 | self.assertEquals(c2.children, []) | |
84 | self.assertEquals(c3.children, [c4]) |
|
84 | self.assertEquals(c3.children, [c4]) | |
85 | self.assertEquals(c4.children, []) |
|
85 | self.assertEquals(c4.children, []) | |
86 |
|
86 | |||
87 | def test_root(self): |
|
87 | def test_root(self): | |
88 | c1 = Component(None) |
|
88 | c1 = Component(None) | |
89 | c2 = Component(c1) |
|
89 | c2 = Component(c1) | |
90 | c3 = Component(c1) |
|
90 | c3 = Component(c1) | |
91 | c4 = Component(c3) |
|
91 | c4 = Component(c3) | |
92 | self.assertEquals(c1.root, c1.root) |
|
92 | self.assertEquals(c1.root, c1.root) | |
93 | self.assertEquals(c2.root, c1) |
|
93 | self.assertEquals(c2.root, c1) | |
94 | self.assertEquals(c3.root, c1) |
|
94 | self.assertEquals(c3.root, c1) | |
95 | self.assertEquals(c4.root, c1) |
|
95 | self.assertEquals(c4.root, c1) | |
96 |
|
96 | |||
97 | def test_change_parent(self): |
|
97 | def test_change_parent(self): | |
98 | c1 = Component(None) |
|
98 | c1 = Component(None) | |
99 | c2 = Component(None) |
|
99 | c2 = Component(None) | |
100 | c3 = Component(c1) |
|
100 | c3 = Component(c1) | |
101 | self.assertEquals(c3.root, c1) |
|
101 | self.assertEquals(c3.root, c1) | |
102 | self.assertEquals(c3.parent, c1) |
|
102 | self.assertEquals(c3.parent, c1) | |
103 | self.assertEquals(c1.children,[c3]) |
|
103 | self.assertEquals(c1.children,[c3]) | |
104 | c3.parent = c2 |
|
104 | c3.parent = c2 | |
105 | self.assertEquals(c3.root, c2) |
|
105 | self.assertEquals(c3.root, c2) | |
106 | self.assertEquals(c3.parent, c2) |
|
106 | self.assertEquals(c3.parent, c2) | |
107 | self.assertEquals(c2.children,[c3]) |
|
107 | self.assertEquals(c2.children,[c3]) | |
108 | self.assertEquals(c1.children,[]) |
|
108 | self.assertEquals(c1.children,[]) | |
109 |
|
109 | |||
110 | def test_subclass_parent(self): |
|
110 | def test_subclass_parent(self): | |
111 | c1 = Component(None) |
|
111 | c1 = Component(None) | |
112 | self.assertRaises(TraitletError, setattr, c1, 'parent', 10) |
|
112 | self.assertRaises(TraitletError, setattr, c1, 'parent', 10) | |
113 |
|
113 | |||
114 | class MyComponent(Component): |
|
114 | class MyComponent(Component): | |
115 | pass |
|
115 | pass | |
116 | c1 = Component(None) |
|
116 | c1 = Component(None) | |
117 | c2 = MyComponent(c1) |
|
117 | c2 = MyComponent(c1) | |
118 | self.assertEquals(MyComponent.parent.this_class, Component) |
|
118 | self.assertEquals(MyComponent.parent.this_class, Component) | |
119 | self.assertEquals(c2.parent, c1) |
|
119 | self.assertEquals(c2.parent, c1) | |
120 |
|
120 | |||
121 | def test_bad_root(self): |
|
121 | def test_bad_root(self): | |
122 | c1 = Component(None) |
|
122 | c1 = Component(None) | |
123 | c2 = Component(None) |
|
123 | c2 = Component(None) | |
124 | c3 = Component(None) |
|
124 | c3 = Component(None) | |
125 | self.assertRaises(ComponentError, setattr, c1, 'root', c2) |
|
125 | self.assertRaises(ComponentError, setattr, c1, 'root', c2) | |
126 | c1.parent = c2 |
|
126 | c1.parent = c2 | |
127 | self.assertEquals(c1.root, c2) |
|
127 | self.assertEquals(c1.root, c2) | |
128 | self.assertRaises(ComponentError, setattr, c1, 'root', c3) |
|
128 | self.assertRaises(ComponentError, setattr, c1, 'root', c3) | |
129 |
|
129 | |||
130 |
|
130 | |||
131 | class TestComponentConfig(TestCase): |
|
131 | class TestComponentConfig(TestCase): | |
132 |
|
132 | |||
133 | def test_default(self): |
|
133 | def test_default(self): | |
134 | c1 = Component(None) |
|
134 | c1 = Component(None) | |
135 | c2 = Component(c1) |
|
135 | c2 = Component(c1) | |
136 | c3 = Component(c2) |
|
136 | c3 = Component(c2) | |
137 | self.assertEquals(c1.config, c2.config) |
|
137 | self.assertEquals(c1.config, c2.config) | |
138 | self.assertEquals(c2.config, c3.config) |
|
138 | self.assertEquals(c2.config, c3.config) | |
139 |
|
139 | |||
140 | def test_custom(self): |
|
140 | def test_custom(self): | |
141 | config = Config() |
|
141 | config = Config() | |
142 | config.foo = 'foo' |
|
142 | config.foo = 'foo' | |
143 | config.bar = 'bar' |
|
143 | config.bar = 'bar' | |
144 | c1 = Component(None, config=config) |
|
144 | c1 = Component(None, config=config) | |
145 | c2 = Component(c1) |
|
145 | c2 = Component(c1) | |
146 | c3 = Component(c2) |
|
146 | c3 = Component(c2) | |
147 | self.assertEquals(c1.config, config) |
|
147 | self.assertEquals(c1.config, config) | |
148 | self.assertEquals(c2.config, config) |
|
148 | self.assertEquals(c2.config, config) | |
149 | self.assertEquals(c3.config, config) |
|
149 | self.assertEquals(c3.config, config) | |
150 | # Test that we always make copies |
|
150 | # Test that we always make copies | |
151 | self.assert_(c1.config is not config) |
|
151 | self.assert_(c1.config is not config) | |
152 | self.assert_(c2.config is not config) |
|
152 | self.assert_(c2.config is not config) | |
153 | self.assert_(c3.config is not config) |
|
153 | self.assert_(c3.config is not config) | |
154 | self.assert_(c1.config is not c2.config) |
|
154 | self.assert_(c1.config is not c2.config) | |
155 | self.assert_(c2.config is not c3.config) |
|
155 | self.assert_(c2.config is not c3.config) | |
156 |
|
156 | |||
157 | def test_inheritance(self): |
|
157 | def test_inheritance(self): | |
158 | class MyComponent(Component): |
|
158 | class MyComponent(Component): | |
159 | a = Int(1, config=True) |
|
159 | a = Int(1, config=True) | |
160 | b = Float(1.0, config=True) |
|
160 | b = Float(1.0, config=True) | |
161 | c = Str('no config') |
|
161 | c = Str('no config') | |
162 | config = Config() |
|
162 | config = Config() | |
163 | config.MyComponent.a = 2 |
|
163 | config.MyComponent.a = 2 | |
164 | config.MyComponent.b = 2.0 |
|
164 | config.MyComponent.b = 2.0 | |
165 | c1 = MyComponent(None, config=config) |
|
165 | c1 = MyComponent(None, config=config) | |
166 | c2 = MyComponent(c1) |
|
166 | c2 = MyComponent(c1) | |
167 | self.assertEquals(c1.a, config.MyComponent.a) |
|
167 | self.assertEquals(c1.a, config.MyComponent.a) | |
168 | self.assertEquals(c1.b, config.MyComponent.b) |
|
168 | self.assertEquals(c1.b, config.MyComponent.b) | |
169 | self.assertEquals(c2.a, config.MyComponent.a) |
|
169 | self.assertEquals(c2.a, config.MyComponent.a) | |
170 | self.assertEquals(c2.b, config.MyComponent.b) |
|
170 | self.assertEquals(c2.b, config.MyComponent.b) | |
171 | c4 = MyComponent(c2, config=Config()) |
|
171 | c4 = MyComponent(c2, config=Config()) | |
172 | self.assertEquals(c4.a, 1) |
|
172 | self.assertEquals(c4.a, 1) | |
173 | self.assertEquals(c4.b, 1.0) |
|
173 | self.assertEquals(c4.b, 1.0) | |
174 |
|
174 | |||
|
175 | def test_parent(self): | |||
|
176 | class Foo(Component): | |||
|
177 | a = Int(0, config=True) | |||
|
178 | b = Str('nope', config=True) | |||
|
179 | class Bar(Foo): | |||
|
180 | b = Str('gotit', config=False) | |||
|
181 | c = Float(config=True) | |||
|
182 | config = Config() | |||
|
183 | config.Foo.a = 10 | |||
|
184 | config.Foo.b = "wow" | |||
|
185 | config.Bar.b = 'later' | |||
|
186 | config.Bar.c = 100.0 | |||
|
187 | f = Foo(None, config=config) | |||
|
188 | b = Bar(f) | |||
|
189 | self.assertEquals(f.a, 10) | |||
|
190 | self.assertEquals(f.b, 'wow') | |||
|
191 | self.assertEquals(b.b, 'gotit') | |||
|
192 | self.assertEquals(b.c, 100.0) | |||
|
193 | ||||
|
194 | ||||
175 | class TestComponentName(TestCase): |
|
195 | class TestComponentName(TestCase): | |
176 |
|
196 | |||
177 | def test_default(self): |
|
197 | def test_default(self): | |
178 | class MyComponent(Component): |
|
198 | class MyComponent(Component): | |
179 | pass |
|
199 | pass | |
180 | c1 = Component(None) |
|
200 | c1 = Component(None) | |
181 | c2 = MyComponent(None) |
|
201 | c2 = MyComponent(None) | |
182 | c3 = Component(c2) |
|
202 | c3 = Component(c2) | |
183 | self.assertNotEquals(c1.name, c2.name) |
|
203 | self.assertNotEquals(c1.name, c2.name) | |
184 | self.assertNotEquals(c1.name, c3.name) |
|
204 | self.assertNotEquals(c1.name, c3.name) | |
185 |
|
205 | |||
186 | def test_manual(self): |
|
206 | def test_manual(self): | |
187 | class MyComponent(Component): |
|
207 | class MyComponent(Component): | |
188 | pass |
|
208 | pass | |
189 | c1 = Component(None, name='foo') |
|
209 | c1 = Component(None, name='foo') | |
190 | c2 = MyComponent(None, name='bar') |
|
210 | c2 = MyComponent(None, name='bar') | |
191 | c3 = Component(c2, name='bah') |
|
211 | c3 = Component(c2, name='bah') | |
192 | self.assertEquals(c1.name, 'foo') |
|
212 | self.assertEquals(c1.name, 'foo') | |
193 | self.assertEquals(c2.name, 'bar') |
|
213 | self.assertEquals(c2.name, 'bar') | |
194 | self.assertEquals(c3.name, 'bah') |
|
214 | self.assertEquals(c3.name, 'bah') |
General Comments 0
You need to be logged in to leave comments.
Login now