from guppy.heapy.test import support class FirstCase(support.TestCase): def setUp(self): support.TestCase.setUp(self) self.OH = self.heapy.OutputHandling def test_1(self): class T: def __init__(self, test, numlines, get_num_lines=None, get_more_msg=None): mod = test.OH def get_line_iter(): for i in range(numlines): yield '%d' % i self.mod = mod mod.setup_printing( self, get_line_iter=get_line_iter, max_more_lines=4, get_num_lines=get_num_lines, get_more_msg=get_more_msg) self.aseq(str(T(self, 4)), '0\n1\n2\n3') t = T(self, 6, lambda: 6) self.aseq( str(t), "0\n1\n2\n3\n") x = t.more self.aseq(str(x), '4\n5') self.aseq( str(x.top), "0\n1\n2\n3\n") self.aseq(str(x.all), "0\n1\n2\n3\n4\n5") t = T(self, 6, get_more_msg=lambda f, t: '<%d more rows>' % (6-t)) self.aseq(str(t), '0\n1\n2\n3\n<3 more rows>') def test_main(debug=0): support.run_unittest(FirstCase, debug) if __name__ == "__main__": test_main()