From 0a22a35912d6af9d78b1966be9f4ee87b906d79a Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Wed, 30 Mar 2022 16:35:26 +0200 Subject: [PATCH] refactoring; renaming --- test/unit_tests/classifier_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/unit_tests/classifier_test.py b/test/unit_tests/classifier_test.py index a6a695c..9a34489 100644 --- a/test/unit_tests/classifier_test.py +++ b/test/unit_tests/classifier_test.py @@ -10,8 +10,8 @@ logger.setLevel(logging.DEBUG) @pytest.mark.parametrize("estimator_type", ["mock", "keras"]) @pytest.mark.parametrize("label_format", ["index", "probability"]) -def test_predict(classifier, input_batch, expected_predictions_mapped): - predictions = classifier.predict(input_batch) +def test_classifier(classifier, input_batch, expected_predictions_mapped): + predictions = classifier(input_batch) assert predictions == expected_predictions_mapped