tests/toolchains/python_toolchain_test.py

import json
import os
import pathlib
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(".", "_"))
        self.assertIn(expected, settings["toolchain_label"], str(settings))

        actual = "{v.major}.{v.minor}.{v.micro}".format(v=sys.version_info)
        self.assertEqual(actual, expect_version)


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