def test_dictree(self):
"""dictree has known output (None)"""
a = {'a':1, 'b':2, 'c':{'aa':11, 'bb':22}}
realstdout = sys.stdout
output = StringIO.StringIO()
sys.stdout = output
self.assertEqual(tb.dictree(a), None)
self.assertEqual(tb.dictree(a, attrs=True), None)
self.assertEqual(tb.dictree(a, verbose=True), None)
sys.stdout = realstdout
result = output.getvalue()
output.close()
self.assertRaises(TypeError, tb.dictree, 'bad')
expected = """+
|____a
|____b
|____c
|____aa
|____bb
+
|____a
|____b
|____c
|____aa
|____bb
+
|____a (int)
|____b (int)
|____c (dict [2])
|____aa (int)
|____bb (int)
"""
self.assertEqual(expected, result)
def test_dictree(self):
"""dictree has known output (None)"""
a = {'a':1, 'b':2, 'c':{'aa':11, 'bb':22}}
realstdout = sys.stdout
output = StringIO.StringIO()
sys.stdout = output
self.assertEqual(tb.dictree(a), None)
self.assertEqual(tb.dictree(a, attrs=True), None)
self.assertEqual(tb.dictree(a, verbose=True), None)
sys.stdout = realstdout
result = output.getvalue()
output.close()
self.assertRaises(TypeError, tb.dictree, 'bad')
expected = """+
|____a
|____b
|____c
|____aa
|____bb
+
|____a
|____b
|____c
|____aa
|____bb
+
|____a (int)
|____b (int)
|____c (dict [2])
|____aa (int)
|____bb (int)
"""
self.assertEqual(expected, result)
def __repr__(self):
tb.dictree(self, verbose=True, attrs=True)
return ''
def __repr__(self):
tb.dictree(self, verbose=True, attrs=True)
return ''
