test: correct implementation of the get_matrix_id mock

AntaresSimulatorTeam · Mar 25, 2024 · dc0e93f · dc0e93f
1 parent 70dc017
commit dc0e93f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/tests/variantstudy/conftest.py b/tests/variantstudy/conftest.py
@@ -81,7 +81,11 @@ def get_matrix_id(matrix: t.Union[t.List[t.List[float]], str]) -> str:
         Get the matrix ID from a matrix or a matrix link.
         """
         if isinstance(matrix, str):
-            return matrix.lstrip("matrix://")
+            # str.removeprefix() is not available in Python 3.8
+            prefix = "matrix://"
+            if matrix.startswith(prefix):
+                return matrix[len(prefix) :]
+            return matrix
         elif isinstance(matrix, list):
             return create(matrix)
         else:

diff --git a/tests/variantstudy/test_command_factory.py b/tests/variantstudy/test_command_factory.py
@@ -403,7 +403,11 @@ def setup_class(self):
     @pytest.mark.unit_test
     def test_command_factory(self, command_dto: CommandDTO):
         def get_matrix_id(matrix: str) -> str:
-            return matrix.lstrip("matrix://")
+            # str.removeprefix() is not available in Python 3.8
+            prefix = "matrix://"
+            if matrix.startswith(prefix):
+                return matrix[len(prefix) :]
+            return matrix
 
         command_factory = CommandFactory(
             generator_matrix_constants=Mock(spec=GeneratorMatrixConstants),