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 ''