Current File : /opt/cloudlinux/venv/lib/python3.11/site-packages/guppy/heapy/test//test_OutputHandling.py
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<Lines 0..3 of 6. Type e.g. '_.more' for more.>")
x = t.more
self.aseq(str(x), '4\n5')
self.aseq(
str(x.top), "0\n1\n2\n3\n<Lines 0..3 of 6. Type e.g. '_.more' for more.>")
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()
