golden hour

/opt/alt/python37/lib64/python3.7/idlelib/idle_test

File/Folder	Size	Actions
README.txt	8.52 KB
__init__.py	712 B
__pycache__	-
htest.py	14.83 KB
mock_idle.py	1.9 KB
mock_tk.py	11.35 KB
template.py	642 B
test_autocomplete.py	10.47 KB
test_autocomplete_w.py	709 B
test_autoexpand.py	4.53 KB
test_browser.py	7.78 KB
test_calltip.py	9.7 KB
test_calltip_w.py	686 B
test_codecontext.py	15.74 KB
test_colorizer.py	14.67 KB
test_config.py	31.29 KB
test_config_key.py	9.48 KB
test_configdialog.py	53.08 KB
test_debugger.py	571 B
test_debugger_r.py	631 B
test_debugobj.py	1.52 KB
test_debugobj_r.py	545 B
test_delegator.py	1.53 KB
test_editmenu.py	2.5 KB
test_editor.py	7.35 KB
test_filelist.py	795 B
test_format.py	23.06 KB
test_grep.py	4.95 KB
test_help.py	849 B
test_help_about.py	5.78 KB
test_history.py	5.39 KB
test_hyperparser.py	8.87 KB
test_iomenu.py	1.25 KB
test_macosx.py	3.23 KB
test_mainmenu.py	594 B
test_multicall.py	1.29 KB
test_outwin.py	5.42 KB
test_parenmatch.py	3.47 KB
test_pathbrowser.py	2.37 KB
test_percolator.py	3.97 KB
test_pyparse.py	18.78 KB
test_pyshell.py	2.12 KB
test_query.py	14.92 KB
test_redirector.py	4.08 KB
test_replace.py	8.11 KB
test_rpc.py	805 B
test_run.py	11.46 KB
test_runscript.py	777 B
test_scrolledlist.py	496 B
test_search.py	2.4 KB
test_searchbase.py	5.5 KB
test_searchengine.py	11.27 KB
test_sidebar.py	12.92 KB
test_squeezer.py	19.62 KB
test_stackviewer.py	1.18 KB
test_statusbar.py	1.11 KB
test_text.py	6.81 KB
test_textview.py	7.19 KB
test_tooltip.py	5.26 KB
test_tree.py	1.71 KB
test_undo.py	4.13 KB
test_warning.py	2.68 KB
test_window.py	1.05 KB
test_zoomheight.py	999 B

Edit: test_warning.py

'''Test warnings replacement in pyshell.py and run.py.

This file could be expanded to include traceback overrides
(in same two modules). If so, change name.
Revise if output destination changes (http://bugs.python.org/issue18318).
Make sure warnings module is left unaltered (http://bugs.python.org/issue18081).
'''
from idlelib import run
from idlelib import pyshell as shell
import unittest
from test.support import captured_stderr
import warnings

# Try to capture default showwarning before Idle modules are imported.
showwarning = warnings.showwarning
# But if we run this file within idle, we are in the middle of the run.main loop
# and default showwarnings has already been replaced.
running_in_idle = 'idle' in showwarning.__name__

# The following was generated from pyshell.idle_formatwarning
# and checked as matching expectation.
idlemsg = '''
Warning (from warnings module):
  File "test_warning.py", line 99
    Line of code
UserWarning: Test
'''
shellmsg = idlemsg + ">>> "

class RunWarnTest(unittest.TestCase):

@unittest.skipIf(running_in_idle, "Does not work when run within Idle.")
    def test_showwarnings(self):
        self.assertIs(warnings.showwarning, showwarning)
        run.capture_warnings(True)
        self.assertIs(warnings.showwarning, run.idle_showwarning_subproc)
        run.capture_warnings(False)
        self.assertIs(warnings.showwarning, showwarning)

def test_run_show(self):
        with captured_stderr() as f:
            run.idle_showwarning_subproc(
                    'Test', UserWarning, 'test_warning.py', 99, f, 'Line of code')
            # The following uses .splitlines to erase line-ending differences
            self.assertEqual(idlemsg.splitlines(), f.getvalue().splitlines())

class ShellWarnTest(unittest.TestCase):

@unittest.skipIf(running_in_idle, "Does not work when run within Idle.")
    def test_showwarnings(self):
        self.assertIs(warnings.showwarning, showwarning)
        shell.capture_warnings(True)
        self.assertIs(warnings.showwarning, shell.idle_showwarning)
        shell.capture_warnings(False)
        self.assertIs(warnings.showwarning, showwarning)

def test_idle_formatter(self):
        # Will fail if format changed without regenerating idlemsg
        s = shell.idle_formatwarning(
                'Test', UserWarning, 'test_warning.py', 99, 'Line of code')
        self.assertEqual(idlemsg, s)

def test_shell_show(self):
        with captured_stderr() as f:
            shell.idle_showwarning(
                    'Test', UserWarning, 'test_warning.py', 99, f, 'Line of code')
            self.assertEqual(shellmsg.splitlines(), f.getvalue().splitlines())

if __name__ == '__main__':
    unittest.main(verbosity=2)