Adding tests for AI pytorch container

Author: cjainsuse

bci_tester/data.py

@@ -1026,6 +1026,13 @@ def create_BCI(
     custom_entry_point="/bin/bash",
 )
 
+PYTORCH_CONTAINER = create_BCI(
+    build_tag=f"{SAC_CONTAINER_PREFIX}/pytorch:2.5",
+    bci_type=ImageType.SAC_APPLICATION,
+    available_versions=["15.6-ai"],
+    custom_entry_point="/bin/bash",
+)
+
 
 STUNNEL_CONTAINER = create_BCI(
     build_tag=f"{APP_CONTAINER_PREFIX}/stunnel:5",
@@ -1101,6 +1108,7 @@ def create_BCI(
     MINIMAL_CONTAINER,
     OLLAMA_CONTAINER,
     MILVUS_CONTAINER,
+    PYTORCH_CONTAINER,
     *POSTFIX_CONTAINERS,
     *TOMCAT_CONTAINERS,
     *POSTGRESQL_CONTAINERS,
@@ -1145,6 +1153,7 @@ def create_BCI(
             OLLAMA_CONTAINER,
             OPENWEBUI_CONTAINER,
             MILVUS_CONTAINER,
+            PYTORCH_CONTAINER,
         ]
         + BASE_FIPS_CONTAINERS
         + CONTAINER_389DS_CONTAINERS

pyproject.toml

@@ -142,6 +142,7 @@ markers = [
     'python_3.12',
     'python_3.13',
     'python_3.6',
+    'pytorch_2.5',
     'registry_2.8',
     'ruby_2.5',
     'ruby_latest',

tests/test_ai.py

@@ -9,8 +9,14 @@
 from bci_tester.data import MILVUS_CONTAINER
 from bci_tester.data import OLLAMA_CONTAINER
 from bci_tester.data import OPENWEBUI_CONTAINER
+from bci_tester.data import PYTORCH_CONTAINER
 
-CONTAINER_IMAGES = (OLLAMA_CONTAINER, OPENWEBUI_CONTAINER, MILVUS_CONTAINER)
+CONTAINER_IMAGES = (
+    OLLAMA_CONTAINER,
+    OPENWEBUI_CONTAINER,
+    MILVUS_CONTAINER,
+    PYTORCH_CONTAINER,
+)
 
 
 @pytest.mark.parametrize(
@@ -71,3 +77,18 @@ def test_milvus_health(container_per_test):
     )
     container_per_test.connection.check_output("etcd --version")
     container_per_test.connection.check_output("milvus")
+
+
+@pytest.mark.parametrize(
+    "container_per_test",
+    [PYTORCH_CONTAINER],
+    indirect=["container_per_test"],
+)
+def test_pytorch_health(container_per_test):
+    """Test the pytorch container."""
+
+    # chech pytorch Version
+    container_per_test.connection.check_output(
+        "python3.11 -c 'import torch; print(torch.__version__)'"
+    )
+    container_per_test.connection.check_output("git --version")

tests/test_metadata.py

@@ -83,6 +83,7 @@
 from bci_tester.data import POSTGRESQL_CONTAINERS
 from bci_tester.data import PROMETHEUS_CONTAINERS
 from bci_tester.data import PYTHON_CONTAINERS
+from bci_tester.data import PYTORCH_CONTAINER
 from bci_tester.data import RUBY_CONTAINERS
 from bci_tester.data import RUST_CONTAINERS
 from bci_tester.data import SAC_PYTHON_CONTAINERS
@@ -276,6 +277,7 @@ def _get_container_label_prefix(
         (OLLAMA_CONTAINER, "ollama", ImageType.SAC_APPLICATION),
         (OPENWEBUI_CONTAINER, "open-webui", ImageType.SAC_APPLICATION),
         (MILVUS_CONTAINER, "milvus", ImageType.SAC_APPLICATION),
+        (PYTORCH_CONTAINER, "pytorch", ImageType.SAC_APPLICATION),
     ]
     + [(STUNNEL_CONTAINER, "stunnel", ImageType.APPLICATION)]
     + [