obmc/utils/testpathtree.py

8         pt['/a'] = 1
12         pt['/a/b'] = 2
21         pt['/a'] = 1
22         self.assertEqual(1, pt['/a'])
26         pt['/a/b'] = 2
27         self.assertEqual(set(['/a', '/a/b']), set(pt.keys()))
28         self.assertEqual(2, pt['/a/b'])
31         self.assertEquals(1, PathTree().get('/a', 1))
35         pt['/a'] = 1
36         self.assertEquals(1, pt.get('/a'))
40         pt['/a'] = 1.1
42         self.assertEqual(set(['/a', '/b']), set(pt.keys()))
43         self.assertEqual(1.1, pt['/a'])
48         pt['/a/b'] = 1.1
49         pt['/a/c'] = 1.2
50         self.assertEqual(set(['/a', '/a/b', '/a/c']), set(pt.keys()))
51         self.assertEqual(1.1, pt['/a/b'])
52         self.assertEqual(1.2, pt['/a/c'])
56         pt['/a/b'] = 1.1
58         self.assertEqual(set(['/a', '/b', '/a/b', '/b/c']), set(pt.keys()))
59         self.assertEqual(1.1, pt['/a/b'])
68         pt['/a/b'] = { 1 : 1.1 }
72         self.assertEquals(set(['a', 'b']), set(dump['/'].keys()))
73         self.assertEquals(set(['b']), set(dump['/']['a'].keys()))
78         pt['/a'] = 1
79         del pt['/a']
84         pt['/a'] = 1
86         del pt['/a']
91         pt['/a/b'] = 1
93         del pt['/a/b']
98         pt['/a'] = 1
99         self.assertEquals(1, pt.setdefault('/a', 2))
102         self.assertEquals(1, PathTree().setdefault('/a', 1))
106             del PathTree()['/a']
110         pt['/a'] = 1
111         self.assertEquals(set([1]), set(pt.values()))
115         pt['/a'] = 1
117         self.assertEquals(set([1, 2]), set(pt.values()))
121         pt['/a'] = 1
122         self.assertEquals(set([('/a', 1)]), set(pt.items()))
126         pt['/a'] = 1
128         self.assertEquals(set([('/a', 1), ('/b', 2)]), set(pt.items()))
132         pt['/a/b'] = 1
133         self.assertEquals(set([('/a', None), ('/a/b', 1)]), set(pt.items()))
141         pt['/b'] = 1
142         self.assertEquals(set([('/b', 1)]), set(pt.dataitems()))
146         pt['/a'] = 1
148         self.assertEquals(set([('/a', 1), ('/b/c', 2)]), set(pt.dataitems()))
152         pt['/a'] = 1
153         self.assertEquals(set(['a']), set(pt.get_children('/')))
154         self.assertEquals(set(), set(pt.get_children('/a')))
158         pt['/a/b'] = 1
159         self.assertEquals(set(['a']), set(pt.get_children('/')))
160         self.assertEquals(set(['b']), set(pt.get_children('/a')))
161         self.assertEquals(set(), set(pt.get_children('/a/b')))
165         pt['/a'] = 1
166         self.assertEquals([1], pt.values())
167         pt.demote('/a')
172         pt['/a'] = 1
174         self.assertEquals(set([1, 2]), set(pt.values()))
175         pt.demote('/a')
180         pt['/a'] = 1
181         pt['/a/b'] = 2
182         self.assertEquals(set([1, 2]), set(pt.values()))
183         pt.demote('/a/b')
184         self.assertEquals(set([1, None]), set(pt.values()))
188         pt['/a'] = 1
191         self.assertEquals('/a', k)
192         self.assertEquals(1, v)
198         pt['/a'] = 1
202         self.assertEquals('/a', k)
203         self.assertEquals(1, v)
212         pt['/a'] = 1
213         pt['/a/b'] = 2
216         self.assertEquals('/a', k)
217         self.assertEquals(1, v)
219         self.assertEquals('/a/b', k)
226         pt['/a'] = 1
227         pt['/a/b'] = 2
228         self.assertEquals(set(['/a']), set(pt.keys(depth=1)))
232         pt['/a'] = 1
233         pt['/a/b'] = 2
234         self.assertEquals(set([1]), set(pt.values(depth=1)))
238         pt['/a'] = 1
239         pt['/a/b'] = 2
240         self.assertEquals(set([('/a', 1)]), set(pt.items(depth=1)))
244         pt['/a'] = 1
245         pt['/a/b'] = 2
247         pt['/b/c'] = 1
248         self.assertEquals(set([('/a', 1)]), set(pt.dataitems(depth=1)))
252         pt['/a'] = 1
253         pt['/a/b'] = 2
255         self.assertEquals(set(['/a/b']), set(pt.keys(subtree='/a')))
259         pt['/a'] = 1
260         pt['/a/b'] = 2
262         self.assertEquals(set([2]), set(pt.values(subtree='/a')))
266         pt['/a'] = 1
267         pt['/a/b'] = 2
269         self.assertEquals(set([('/a/b', 2)]), set(pt.items(subtree='/a')))
273         pt['/a'] = 1
274         pt['/a/b'] = 2
275         pt['/a/c'] = None
277         self.assertEquals(set([('/a/b', 2)]), set(pt.dataitems(subtree='/a')))
281         pt['/a'] = 1
282         pt['/a/b'] = 2
283         pt['/a/b/c'] = 3
285         self.assertEquals(set(['/a/b']), set(pt.keys(subtree='/a', depth=1)))
289         pt['/a'] = 1
290         pt['/a/b'] = 2
291         pt['/a/b/c'] = 3
293         self.assertEquals(set([2]), set(pt.values(subtree='/a', depth=1)))
297         pt['/a'] = 1
298         pt['/a/b'] = 2
299         pt['/a/b/c'] = 3
301         self.assertEquals(set([('/a/b', 2)]), set(pt.items(subtree='/a', depth=1)))
305         pt['/a'] = 1
306         pt['/a/b'] = 2
307         pt['/a/b/c'] = 3
308         pt['/a/c'] = None
310         self.assertEquals(set([('/a/b', 2)]), set(pt.dataitems(subtree='/a', depth=1)))
316     pt[k] = 1
317     pt[k] = pt[k] + 1
324 key = '/' + '/'.join(['a'] * {})
335     pt["/{}/a".format(i)] = i
336 key = '/0/a'
347 for i in ('/' + '/'.join([chr(ord('a') + x)] * {}) for x in range(0, 26)):
355     for depth in range(1, 11):
370     for depth in range(1, 5):
371         for width in range(1, 5):