From fcf6e8e98eff354bc4dd3f0ac1c0eff862be5e6a Mon Sep 17 00:00:00 2001 From: Tom Ritchford Date: Tue, 15 Sep 2020 17:22:40 +0200 Subject: [PATCH] Test print argument to lark.tools.standalone.main --- tests/test_tools.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/tests/test_tools.py b/tests/test_tools.py index e691237..f7e0c47 100644 --- a/tests/test_tools.py +++ b/tests/test_tools.py @@ -1,8 +1,9 @@ -from __future__ import absolute_import +from __future__ import absolute_import, print_function import sys from unittest import TestCase, main +from functools import partial from lark.tree import Tree from lark.tools import standalone @@ -20,10 +21,8 @@ class TestStandalone(TestCase): def _create_standalone(self, grammar): code_buf = StringIO() - temp = sys.stdout - sys.stdout = code_buf - standalone.main(StringIO(grammar), 'start') - sys.stdout = temp + pr = partial(print, file=code_buf) + standalone.main(StringIO(grammar), 'start', print=pr) code = code_buf.getvalue() context = {} @@ -137,5 +136,3 @@ class TestStandalone(TestCase): if __name__ == '__main__': main() - -