pipelines/components/aws/sagemaker/tests/unit_tests
Nicholas Thomson ddd1969b34
[AWS SageMaker] Unit tests for Training component (#3722)
* Added additional training unit tests

* Add main training function tests

* Add full training test coverage

* Fix import sys

* Fix poorly named test
2020-05-13 16:14:22 -07:00
..
tests [AWS SageMaker] Unit tests for Training component (#3722) 2020-05-13 16:14:22 -07:00
.gitignore AWS Sagemaker : Add unit tests (#3642) 2020-04-30 01:32:18 -07:00
Dockerfile AWS Sagemaker : Add unit tests (#3642) 2020-04-30 01:32:18 -07:00
README.md AWS Sagemaker : Add unit tests (#3642) 2020-04-30 01:32:18 -07:00
pytest.ini [AWS SageMaker] Unit tests for Training component (#3722) 2020-05-13 16:14:22 -07:00
requirements.txt AWS Sagemaker : Add unit tests (#3642) 2020-04-30 01:32:18 -07:00
run_all_tests.sh [AWS SageMaker] Unit tests for Training component (#3722) 2020-05-13 16:14:22 -07:00

README.md

Unit tests for AWS SageMaker KFP Components

How to run these tests

Method 1 : Run these tests using docker

  1. Clone the git repo
    git clone https://github.com/kubeflow/pipelines.git
    
  2. Build the dockerfile
    cd pipelines/components/aws 
    docker build ./ -f ./sagemaker/tests/unit_tests/Dockerfile -t amazon/unit-test-aws-sagemaker-kfp-components
    
  3. Run all unit tests
    docker run -it amazon/unit-test-aws-sagemaker-kfp-components
    

Method 2 : Run these tests locally

  1. Clone the git repo
    git clone https://github.com/kubeflow/pipelines.git
    
  2. Install the pip packages required for testing
    cd pipelines/components/aws/sagemaker/tests/unit_tests/
    
    pip install -r requirements.txt 
    
  3. Run all unit tests
    # while in the same directory pipelines/components/aws/sagemaker/tests/unit_tests/
    
    ./run_all_tests.sh