Better tests directory structure (#2231)

tests/image_specific_tests/docker-stacks-foundation/test_python_version.py

@@ -0,0 +1,34 @@
+# Copyright (c) Jupyter Development Team.
+# Distributed under the terms of the Modified BSD License.
+import logging
+
+from tests.utils.tracked_container import TrackedContainer
+
+LOGGER = logging.getLogger(__name__)
+EXPECTED_PYTHON_VERSION = "3.12"
+
+
+def test_python_version(container: TrackedContainer) -> None:
+    LOGGER.info(
+        f"Checking that python major.minor version is {EXPECTED_PYTHON_VERSION}"
+    )
+    logs = container.run_and_wait(
+        timeout=5,
+        tty=True,
+        command=["python", "--version"],
+    )
+    python = next(line for line in logs.splitlines() if line.startswith("Python "))
+    full_version = python.split()[1]
+    major_minor_version = full_version[: full_version.rfind(".")]
+
+    assert major_minor_version == EXPECTED_PYTHON_VERSION
+
+
+def test_python_pinned_version(container: TrackedContainer) -> None:
+    LOGGER.info(f"Checking that pinned python version is {EXPECTED_PYTHON_VERSION}.*")
+    logs = container.run_and_wait(
+        timeout=5,
+        tty=True,
+        command=["cat", "/opt/conda/conda-meta/pinned"],
+    )
+    assert f"python {EXPECTED_PYTHON_VERSION}.*" in logs