From ab2beaf1a0ff5a92398d324398cf80f24cd9e260 Mon Sep 17 00:00:00 2001 From: MOSES OLAFENWA Date: Thu, 15 Aug 2019 13:09:51 +0100 Subject: [PATCH] Code coverage integration --- .codecov.yml | 2 +- .travis.yml | 2 +- test/test_custom_recognition.py | 12 ++++++------ test/test_model_training.py | 4 ++++ 4 files changed, 12 insertions(+), 8 deletions(-) diff --git a/.codecov.yml b/.codecov.yml index 2b5037ed..d580401b 100644 --- a/.codecov.yml +++ b/.codecov.yml @@ -5,7 +5,7 @@ codecov: coverage: precision: 2 round: down - range: "70...100" + range: "30...100" status: project: yes diff --git a/.travis.yml b/.travis.yml index 46e9a99a..0d1fc18f 100644 --- a/.travis.yml +++ b/.travis.yml @@ -25,7 +25,7 @@ script: - wget -P data-models/ https://github.com/OlafenwaMoses/ImageAI/releases/download/1.0/yolo-tiny.h5 - wget -P data-models/ https://github.com/OlafenwaMoses/ImageAI/releases/download/essential-v4/pretrained-yolov3.h5 - wget -P data-models/ https://github.com/OlafenwaMoses/ImageAI/releases/download/essential-v4/hololens-ex-60--loss-2.76.h5 - - pytest -v + - pytest -m recognition -v after_success: - codecov - coverage diff --git a/test/test_custom_recognition.py b/test/test_custom_recognition.py index 4b45a2c3..fd113d33 100644 --- a/test/test_custom_recognition.py +++ b/test/test_custom_recognition.py @@ -17,7 +17,7 @@ def images_to_image_array(): - +@pytest.mark.recognition @pytest.mark.resnet @pytest.mark.recognition_custom def test_custom_recognition_model_resnet(): @@ -35,7 +35,7 @@ def test_custom_recognition_model_resnet(): assert isinstance(predictions[0], str) assert isinstance(probabilities[0], str) - +@pytest.mark.recognition @pytest.mark.resnet @pytest.mark.recognition_custom def test_custom_recognition_full_model_resnet(): @@ -51,7 +51,7 @@ def test_custom_recognition_full_model_resnet(): assert isinstance(predictions[0], str) assert isinstance(probabilities[0], str) - +@pytest.mark.recognition @pytest.mark.densenet @pytest.mark.recognition_custom def test_custom_recognition_model_densenet(): @@ -70,7 +70,7 @@ def test_custom_recognition_model_densenet(): - +@pytest.mark.recognition @pytest.mark.resnet @pytest.mark.recognition_custom @pytest.mark.recognition_multi @@ -93,7 +93,7 @@ def test_custom_recognition_model_resnet_multi(): assert isinstance(result["predictions"][0], str) assert isinstance(result["percentage_probabilities"][0], str) - +@pytest.mark.recognition @pytest.mark.resnet @pytest.mark.recognition_custom @pytest.mark.recognition_multi @@ -115,7 +115,7 @@ def test_custom_recognition_full_model_resnet_multi(): assert isinstance(result["predictions"][0], str) assert isinstance(result["percentage_probabilities"][0], str) - +@pytest.mark.recognition @pytest.mark.densenet @pytest.mark.recognition_custom @pytest.mark.recognition_multi diff --git a/test/test_model_training.py b/test/test_model_training.py index 3c253368..f841a2da 100644 --- a/test/test_model_training.py +++ b/test/test_model_training.py @@ -21,6 +21,7 @@ def clear_keras_session(): @pytest.mark.training @pytest.mark.training_resnet @pytest.mark.resnet +@pytest.mark.recognition def test_resnet_training(clear_keras_session): trainer = ModelTraining() @@ -39,6 +40,7 @@ def test_resnet_training(clear_keras_session): @pytest.mark.training @pytest.mark.training_squeezenet @pytest.mark.squeezenet +@pytest.mark.recognition def test_squeezenet_training(clear_keras_session): trainer = ModelTraining() @@ -57,6 +59,7 @@ def test_squeezenet_training(clear_keras_session): @pytest.mark.training @pytest.mark.training_inception_v3 @pytest.mark.inception_v3 +@pytest.mark.recognition def test_inception_v3_training(clear_keras_session): trainer = ModelTraining() @@ -75,6 +78,7 @@ def test_inception_v3_training(clear_keras_session): @pytest.mark.training @pytest.mark.training_densenet @pytest.mark.densenet +@pytest.mark.recognition def test_densenet_training(clear_keras_session): trainer = ModelTraining()