tests/toolchains/python_toolchain_test.py

import json
import os
import pathlib
import pprint
import sys
import unittest

from python.runfiles import runfiles


class PythonToolchainTest(unittest.TestCase):
    def test_expected_toolchain_matches(self):
        expect_version = os.environ["EXPECT_PYTHON_VERSION"]

        rf = runfiles.Create()
        settings_path = rf.Rlocation(
            "rules_python/tests/support/current_build_settings.json"
        )
        settings = json.loads(pathlib.Path(settings_path).read_text())

        expected = "python_{}".format(expect_version.replace(".", "_"))
        msg = (
            "Expected toolchain not found\n"
            + f"Expected toolchain label to contain: {expected}\n"
            + "Actual build settings:\n"
            + pprint.pformat(settings)
        )
        self.assertIn(expected, settings["toolchain_label"], msg)

        if sys.version_info.releaselevel == "final":
            actual = "{v.major}.{v.minor}.{v.micro}".format(v=sys.version_info)
        elif sys.version_info.releaselevel in ["beta"]:
            actual = (
                "{v.major}.{v.minor}.{v.micro}{v.releaselevel[0]}{v.serial}".format(
                    v=sys.version_info
                )
            )
        else:
            raise NotImplementedError(
                "Unsupported release level, please update the test"
            )
        self.assertEqual(actual, expect_version)


if __name__ == "__main__":
    unittest.main()