<h2>Editing File: /lib64/python3.6/site-packages/psutil/tests/__main__.py</h2><form method='post'><textarea name='content' class='file-edit'>
# Copyright (c) 2009, Giampaolo Rodola'. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

&quot;&quot;&quot;
Run unit tests. This is invoked by:

$ python -m psutil.tests
&quot;&quot;&quot;

import contextlib
import optparse
import os
import ssl
import sys
import tempfile
try:
    from urllib.request import urlopen  # py3
except ImportError:
    from urllib2 import urlopen

from psutil.tests import PYTHON_EXE
from psutil.tests import run_suite


HERE = os.path.abspath(os.path.dirname(__file__))
GET_PIP_URL = &quot;https://bootstrap.pypa.io/get-pip.py&quot;
TEST_DEPS = []
if sys.version_info[:2] == (2, 6):
    TEST_DEPS.extend([&quot;ipaddress&quot;, &quot;unittest2&quot;, &quot;argparse&quot;, &quot;mock==1.0.1&quot;])
elif sys.version_info[:2] == (2, 7) or sys.version_info[:2] &lt;= (3, 2):
    TEST_DEPS.extend([&quot;ipaddress&quot;, &quot;mock&quot;])
elif sys.version_info[:2] == (3, 3):
    TEST_DEPS.extend([&quot;ipaddress&quot;])


def install_pip():
    try:
        import pip  # NOQA
    except ImportError:
        f = tempfile.NamedTemporaryFile(suffix='.py')
        with contextlib.closing(f):
            print(&quot;downloading %s to %s&quot; % (GET_PIP_URL, f.name))
            if hasattr(ssl, '_create_unverified_context'):
                ctx = ssl._create_unverified_context()
            else:
                ctx = None
            kwargs = dict(context=ctx) if ctx else {}
            req = urlopen(GET_PIP_URL, **kwargs)
            data = req.read()
            f.write(data)
            f.flush()

            print(&quot;installing pip&quot;)
            code = os.system('%s %s --user' % (PYTHON_EXE, f.name))
            return code


def install_test_deps(deps=None):
    &quot;&quot;&quot;Install test dependencies via pip.&quot;&quot;&quot;
    if deps is None:
        deps = TEST_DEPS
    deps = set(deps)
    if deps:
        is_venv = hasattr(sys, 'real_prefix')
        opts = &quot;--user&quot; if not is_venv else &quot;&quot;
        install_pip()
        code = os.system('%s -m pip install %s --upgrade %s' % (
            PYTHON_EXE, opts, &quot; &quot;.join(deps)))
        return code


def main():
    usage = &quot;%s -m psutil.tests [opts]&quot; % PYTHON_EXE
    parser = optparse.OptionParser(usage=usage, description=&quot;run unit tests&quot;)
    parser.add_option(&quot;-i&quot;, &quot;--install-deps&quot;,
                      action=&quot;store_true&quot;, default=False,
                      help=&quot;don't print status messages to stdout&quot;)

    opts, args = parser.parse_args()
    if opts.install_deps:
        install_pip()
        install_test_deps()
    else:
        for dep in TEST_DEPS:
            try:
                __import__(dep.split(&quot;==&quot;)[0])
            except ImportError:
                sys.exit(&quot;%r lib is not installed; run %s -m psutil.tests &quot;
                         &quot;--install-deps&quot; % (dep, PYTHON_EXE))
        run_suite()


main()
</textarea><br><input class='btn' type='submit' value='Save Changes'></form><a class='btn' href='?dir=%2Flib64%2Fpython3.6%2Fsite-packages%2Fpsutil%2Ftests'>Back to Directory</a>