[tests] Add a test of compiler gym version number.

tests/BUILD

@@ -97,8 +97,10 @@ py_test(
     name = "version_test",
     timeout = "short",
     srcs = ["version_test.py"],
+    data = ["//:VERSION"],
     deps = [
         ":test_main",
         "//compiler_gym",
+        "//tests/pytest_plugins:common",
     ],
 )

tests/pytest_plugins/common.py

@@ -23,6 +23,11 @@
     not sys.platform.lower().startswith("darwin"), reason="macOS only"
 )
 
+# Decorator to mark a test as skipped if not running under bazel.
+bazel_only = pytest.mark.skipif(
+    os.environ.get("TEST_WORKSPACE", "") == "", reason="bazel only"
+)
+
 
 @pytest.fixture(scope="function")
 def tmpwd() -> Path:

tests/version_test.py

@@ -8,7 +8,9 @@
 import pytest
 
 import compiler_gym
+from compiler_gym.util.runfiles_path import runfiles_path
 from packaging import version
+from tests.pytest_plugins.common import bazel_only
 from tests.test_main import main
 
 # Marker to skip a test if running under bazel.
@@ -33,5 +35,13 @@ def test_setuptools_version():
     assert version == compiler_gym.__version__
 
 
+@bazel_only
+def test_expected_version():
+    """Test that embedded compiler gym version matches VERSION file."""
+    with open(runfiles_path("VERSION")) as f:
+        version = f.read().strip()
+    assert version == compiler_gym.__version__
+
+
 if __name__ == "__main__":
     main()