have special assertions for pre releases

Author: aignas

tests/toolchains/defs.bzl

@@ -41,6 +41,11 @@ def define_toolchain_tests(name):
 
         parsed = version.parse(python_version, strict = True)
         expect_python_version = "{0}.{1}.{2}".format(*parsed.release)
+        if parsed.pre:
+            expect_python_version = "{0}{1}{2}".format(
+                expect_python_version,
+                *parsed.pre
+            )
         py_reconfig_test(
             name = "python_{}_test".format(python_version),
             srcs = ["python_toolchain_test.py"],

tests/toolchains/python_toolchain_test.py

@@ -3,11 +3,8 @@
 import pathlib
 import pprint
 import sys
-import textwrap
 import unittest
 
-print("PYTHONPATH is:")
-print(textwrap.indent("\n".join(sys.path), prefix=" " * 4))
 from python.runfiles import runfiles
 
 
@@ -30,7 +27,18 @@ def test_expected_toolchain_matches(self):
         )
         self.assertIn(expected, settings["toolchain_label"], msg)
 
-        actual = "{v.major}.{v.minor}.{v.micro}".format(v=sys.version_info)
+        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)