From c6000a2d839745438ccaa3e1ae383e03aed89bc3 Mon Sep 17 00:00:00 2001 From: RollerKnobster Date: Tue, 25 Jun 2024 11:13:14 +0300 Subject: [PATCH] Fix: adjust test for argo versions --- tests/gordo/workflow/test_helpers.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/gordo/workflow/test_helpers.py b/tests/gordo/workflow/test_helpers.py index 56919193e..78ef29a01 100644 --- a/tests/gordo/workflow/test_helpers.py +++ b/tests/gordo/workflow/test_helpers.py @@ -3,6 +3,7 @@ import pytest from mock import patch from packaging import version +from packaging.version import InvalidVersion from gordo.workflow.workflow_generator.helpers import ( determine_argo_version, @@ -17,7 +18,11 @@ def test_parse_argo_version(): parsed_version = parse_argo_version("2.12.11") assert type(parsed_version) is version.Version assert str(parsed_version) == "2.12.11" - assert parse_argo_version("wrong_version") is None + + +def test_parse_argo_version_error(): + with pytest.raises(InvalidVersion): + parse_argo_version("wrong_version") def create_completed_process(return_code, stdout):