Fix: adjust test for argo versions

equinor · Jun 25, 2024 · c6000a2 · c6000a2
1 parent f138c27
commit c6000a2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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):