Add test for validating anaconda channels (#652)

Author: claytonparnell

test/test_artifacts/v2/anaconda.test.Dockerfile

@@ -0,0 +1,8 @@
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
+
+ARG MAMBA_DOCKERFILE_ACTIVATE=1
+
+COPY --chown=$MAMBA_USER:$MAMBA_USER scripts/run_anaconda_tests.sh .
+RUN chmod +x run_anaconda_tests.sh
+CMD ["./run_anaconda_tests.sh"]

test/test_artifacts/v2/scripts/run_anaconda_tests.sh

@@ -0,0 +1,34 @@
+#!/bin/bash
+
+# Expected output
+expected_output=$'channels:\n  - conda-forge'
+
+# Actual output
+actual_micromamba=$(micromamba config list channels)
+actual_conda=$(conda config --show channels)
+
+# Compare the outputs
+if [ "$actual_micromamba" = "$expected_output" ]; then
+    echo "Micromamba Validation passed: Output matches expected format"
+    exit 0
+else
+    echo "Micromamba Validation failed: Output does not match expected format"
+    echo "Expected:"
+    echo "$expected_output"
+    echo "Actual:"
+    echo "$actual_micromamba"
+    exit 1
+fi
+
+# Compare the outputs
+if [ "$actual_conda" = "$expected_output" ]; then
+    echo "Conda Validation passed: Output matches expected format"
+    exit 0
+else
+    echo "Conda Validation failed: Output does not match expected format"
+    echo "Expected:"
+    echo "$expected_output"
+    echo "Actual:"
+    echo "$actual_conda"
+    exit 1
+fi

test/test_artifacts/v3/anaconda.test.Dockerfile

@@ -0,0 +1,8 @@
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
+
+ARG MAMBA_DOCKERFILE_ACTIVATE=1
+
+COPY --chown=$MAMBA_USER:$MAMBA_USER scripts/run_anaconda_tests.sh .
+RUN chmod +x run_anaconda_tests.sh
+CMD ["./run_anaconda_tests.sh"]

test/test_artifacts/v3/scripts/run_anaconda_tests.sh

@@ -0,0 +1,34 @@
+#!/bin/bash
+
+# Expected output
+expected_output=$'channels:\n  - conda-forge'
+
+# Actual output
+actual_micromamba=$(micromamba config list channels)
+actual_conda=$(conda config --show channels)
+
+# Compare the outputs
+if [ "$actual_micromamba" = "$expected_output" ]; then
+    echo "Micromamba Validation passed: Output matches expected format"
+    exit 0
+else
+    echo "Micromamba Validation failed: Output does not match expected format"
+    echo "Expected:"
+    echo "$expected_output"
+    echo "Actual:"
+    echo "$actual_micromamba"
+    exit 1
+fi
+
+# Compare the outputs
+if [ "$actual_conda" = "$expected_output" ]; then
+    echo "Conda Validation passed: Output matches expected format"
+    exit 0
+else
+    echo "Conda Validation failed: Output does not match expected format"
+    echo "Expected:"
+    echo "$expected_output"
+    echo "Actual:"
+    echo "$actual_conda"
+    exit 1
+fi

test/test_dockerfile_based_harness.py

@@ -18,6 +18,7 @@
 @pytest.mark.parametrize(
     "dockerfile_path, required_packages",
     [
+        ("anaconda.test.Dockerfile", ["conda"]),
         ("keras.test.Dockerfile", ["keras"]),
         ("autogluon.test.Dockerfile", ["autogluon"]),
         ("matplotlib.test.Dockerfile", ["matplotlib"]),
@@ -102,6 +103,7 @@ def test_dockerfiles_for_cpu(
 @pytest.mark.parametrize(
     "dockerfile_path, required_packages",
     [
+        ("anaconda.test.Dockerfile", ["conda"]),
         ("keras.test.Dockerfile", ["keras"]),
         ("autogluon.test.Dockerfile", ["autogluon"]),
         ("matplotlib.test.Dockerfile", ["matplotlib"]),