Fix unit tests (#573)

Co-authored-by: Ruinong Tian <[email protected]>

Author: TRNWWZ

test/test_artifacts/v1/altair.test.Dockerfile

@@ -3,7 +3,7 @@ FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 
-CMD ["python", "-c", "import altair"]
+RUN ["python", "-c", "import altair"]
 
 RUN sudo apt-get update && \
     sudo apt-get install -y git && \

test/test_artifacts/v1/sm-python-sdk.test.Dockerfile

@@ -4,7 +4,7 @@ FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN git clone --recursive https://github.com/aws/sagemaker-python-sdk.git
 
-RUN micromamba install -y -c conda-forge pytest
+RUN micromamba install -y --freeze-installed -c conda-forge pytest mock stopit awslogs
 
 WORKDIR "sagemaker-python-sdk"
 COPY --chown=$MAMBA_USER:$MAMBA_USER scripts/run_pysdk_tests.sh .

test/test_artifacts/v2/altair.test.Dockerfile

@@ -3,7 +3,7 @@ FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 
-CMD ["python", "-c", "import altair"]
+RUN ["python", "-c", "import altair"]
 
 RUN sudo apt-get update && \
     sudo apt-get install -y git && \

test/test_artifacts/v2/sm-python-sdk.test.Dockerfile

@@ -4,6 +4,8 @@ FROM $SAGEMAKER_DISTRIBUTION_IMAGE
 ARG MAMBA_DOCKERFILE_ACTIVATE=1
 RUN git clone --recursive https://github.com/aws/sagemaker-python-sdk.git
 
+RUN micromamba install -y --freeze-installed -c conda-forge pytest mock stopit awslogs
+
 # Sagemaker Python SDK's unit tests requires AWS_DEFAULT_REGION to be set. So, using an arbitrary value of us-east-1
 ENV AWS_DEFAULT_REGION=us-east-1
 WORKDIR "sagemaker-python-sdk"