From 6d6dfa8d36c2aa6b62deac49aef8a06b934accd2 Mon Sep 17 00:00:00 2001 From: Antoine Pitrou Date: Wed, 8 Feb 2012 23:28:36 +0100 Subject: =?UTF-8?q?Issue=20#13609:=20Add=20two=20functions=20to=20query=20?= =?UTF-8?q?the=20terminal=20size:=20os.get=5Fterminal=5Fsize=20(low=20leve?= =?UTF-8?q?l)=20and=20shutil.get=5Fterminal=5Fsize=20(high=20level).=20Pat?= =?UTF-8?q?ch=20by=20Zbigniew=20J=C4=99drzejewski-Szmek.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Lib/test/test_os.py | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) (limited to 'Lib/test/test_os.py') diff --git a/Lib/test/test_os.py b/Lib/test/test_os.py index 4d27c2b314..8dd745a9c7 100644 --- a/Lib/test/test_os.py +++ b/Lib/test/test_os.py @@ -1840,6 +1840,43 @@ class Win32DeprecatedBytesAPI(unittest.TestCase): os.symlink, filename, filename) +@unittest.skipUnless(hasattr(os, 'get_terminal_size'), "requires os.get_terminal_size") +class TermsizeTests(unittest.TestCase): + def test_does_not_crash(self): + """Check if get_terminal_size() returns a meaningful value. + + There's no easy portable way to actually check the size of the + terminal, so let's check if it returns something sensible instead. + """ + try: + size = os.get_terminal_size() + except OSError as e: + if e.errno == errno.EINVAL or sys.platform == "win32": + # Under win32 a generic OSError can be thrown if the + # handle cannot be retrieved + self.skipTest("failed to query terminal size") + raise + + self.assertGreater(size.columns, 0) + self.assertGreater(size.lines, 0) + + def test_stty_match(self): + """Check if stty returns the same results + + stty actually tests stdin, so get_terminal_size is invoked on + stdin explicitly. If stty succeeded, then get_terminal_size() + should work too. + """ + try: + size = subprocess.check_output(['stty', 'size']).decode().split() + except (FileNotFoundError, subprocess.CalledProcessError): + self.skipTest("stty invocation failed") + expected = (int(size[1]), int(size[0])) # reversed order + + actual = os.get_terminal_size(sys.__stdin__.fileno()) + self.assertEqual(expected, actual) + + @support.reap_threads def test_main(): support.run_unittest( @@ -1866,6 +1903,7 @@ def test_main(): ProgramPriorityTests, ExtendedAttributeTests, Win32DeprecatedBytesAPI, + TermsizeTests, ) if __name__ == "__main__": -- cgit v1.2.1