Add Tests For _pythoninfo.py (#147)

Author: ericsnowcurrently

pyperformance/tests/__init__.py

@@ -42,6 +42,26 @@ def run_cmd(*argv):
     print("", flush=True)
 
 
+class Resources:
+    """A mixin that can create resources."""
+
+    def venv(self, python=sys.executable):
+        tmpdir = tempfile.mkdtemp()
+        venv = os.path.join(tmpdir, 'venv')
+        run_cmd(python or sys.executable, '-u', '-m', 'venv', venv)
+        self.addCleanup(lambda: shutil.rmtree(tmpdir))
+        return venv, resolve_venv_python(venv)
+
+
+def resolve_venv_python(venv):
+    if os.name == "nt":
+        basename = os.path.basename(sys.executable)
+        venv_python = os.path.join(venv, 'Scripts', basename)
+    else:
+        venv_python = os.path.join(venv, 'bin', 'python3')
+    return venv_python
+
+
 #############################
 # functional tests
 
@@ -94,11 +114,7 @@ def tearDownClass(cls):
 
     @property
     def venv_python(self):
-        if os.name == "nt":
-            python = os.path.basename(sys.executable)
-            return os.path.join(self._VENV, 'Scripts', python)
-        else:
-            return os.path.join(self._VENV, 'bin', 'python3')
+        return resolve_venv_python(self._VENV)
 
     def run_pyperformance(self, cmd, *args, invenv=True):
         if invenv:

pyperformance/tests/test_pythoninfo.py

@@ -0,0 +1,119 @@
+import importlib.util
+import os
+import sys
+import unittest
+
+from pyperformance import tests, _pythoninfo
+
+
+INFO = {
+    'executable': sys.executable,
+    'version_str': sys.version,
+    'version_info': tuple(sys.version_info),
+    'hexversion': sys.hexversion,
+    'api_version': sys.api_version,
+    # importlib.util.MAGIC_NUMBER has been around since 3.5.
+    'magic_number': importlib.util.MAGIC_NUMBER.hex(),
+    'implementation_name': sys.implementation.name.lower(),
+    'implementation_version': tuple(sys.implementation.version),
+    'platform': sys.platform,
+    'prefix': sys.prefix,
+    'exec_prefix': sys.exec_prefix,
+    'base_prefix': sys.base_prefix,
+    'base_exec_prefix': sys.base_exec_prefix,
+    'platlibdir': getattr(sys, 'platlibdir', 'lib'),
+    'stdlib_dir': os.path.dirname(os.__file__)
+}
+
+
+class GetPythonInfoTests(tests.Resources, unittest.TestCase):
+
+    def test_no_args(self):
+        info = _pythoninfo.get_python_info()
+
+        self.assertEqual(info, INFO)
+
+    def test_current_python(self):
+        info = _pythoninfo.get_python_info(sys.executable)
+
+        self.assertEqual(info, INFO)
+
+    def test_venv(self):
+        self.maxDiff = 80 * 100
+        venv, python = self.venv()
+        expected = dict(INFO)
+        expected['executable'] = python
+        expected['prefix'] = venv
+        expected['exec_prefix'] = venv
+
+        info = _pythoninfo.get_python_info(python)
+
+        self.assertEqual(info, expected)
+
+
+class GetPythonIDTests(unittest.TestCase):
+
+    INFO = {
+        'executable': '/a/b/c/bin/spam-python',
+        'version_str': '3.8.10 (default, May  5 2021, 03:01:07) \n[GCC 7.5.0]',
+        'version_info': (3, 8, 10, 'final', 0),
+        'api_version': 1013,
+        'magic_number': b'U\r\r\n'.hex(),
+        'implementation_name': 'cpython',
+        'implementation_version': (3, 8, 10, 'final', 0),
+    }
+    ID = 'b14d92fd0e6f'
+
+    def test_no_prefix(self):
+        pyid = _pythoninfo.get_python_id(self.INFO)
+
+        self.assertEqual(pyid, self.ID)
+
+    def test_true_prefix(self):
+        pyid = _pythoninfo.get_python_id(self.INFO, prefix=True)
+
+        self.assertEqual(pyid, f'cpython3.8-{self.ID}')
+
+    def test_given_prefix(self):
+        pyid = _pythoninfo.get_python_id(self.INFO, prefix='spam-')
+
+        self.assertEqual(pyid, f'spam-{self.ID}')
+
+
+class InspectPythonInstallTests(tests.Resources, unittest.TestCase):
+
+    BASE = getattr(sys, '_base_executable', None)
+
+    def test_info(self):
+        info = INFO
+        (base, isdev, isvenv,
+         ) = _pythoninfo.inspect_python_install(info)
+
+        if self.BASE:
+            self.assertEqual(base, self.BASE)
+        self.assertEqual(base, info['executable'])
+        self.assertFalse(isdev)
+        self.assertFalse(isvenv)
+
+    def test_normal(self):
+        (base, isdev, isvenv,
+         ) = _pythoninfo.inspect_python_install()
+
+        self.assertEqual(base, sys.executable)
+        self.assertFalse(isdev)
+        self.assertFalse(isvenv)
+
+    def test_venv(self):
+        base_expected = sys.executable
+        _, python = self.venv(base_expected)
+        (base, isdev, isvenv,
+         ) = _pythoninfo.inspect_python_install(python)
+
+        self.assertEqual(base, base_expected)
+        self.assertFalse(isdev)
+        self.assertTrue(isvenv)
+
+    @unittest.skip('needs platform-specific testing')
+    def test_dev(self):
+        # XXX
+        raise NotImplementedError