Implement Model.__repr__

Include the model ID in `Model.__repr__`.

Author: dweindl

petab/v1/models/model.py

@@ -16,6 +16,9 @@ class Model(abc.ABC):
     def __init__(self):
         ...
 
+    def __repr__(self):
+        return f"<{self.__class__.__name__} {self.model_id!r}>"
+
     @staticmethod
     @abc.abstractmethod
     def from_file(filepath_or_buffer: Any, model_id: str) -> Model:

tests/v1/test_model_pysb.py

@@ -87,3 +87,9 @@ def test_pattern_parsing(uses_pysb):
     pattern = pysb.as_complex_pattern(B(s="a") ** c1)
     assert pattern_from_string(str(pattern), model).matches(pattern)
     assert str(pattern) == str(pattern_from_string("B(s='a') ** c1", model))
+
+
+def test_pysb_model_repr(uses_pysb):
+    model = pysb.Model(name="test")
+    petab_model = PySBModel(model)
+    assert repr(petab_model) == "<PySBModel 'test'>"

tests/v1/test_sbml.py

@@ -1,9 +1,12 @@
 import os
 import sys
 
+import libsbml
 import pandas as pd
 import pytest
 
+from petab.v1.models.sbml_model import SbmlModel
+
 sys.path.append(os.getcwd())
 import petab  # noqa: E402
 
@@ -122,3 +125,10 @@ def test_get_condition_specific_models():
         )
 
     check_model(condition_model)
+
+
+def test_sbml_model_repr():
+    sbml_model = libsbml.SBMLDocument().createModel()
+    sbml_model.setId("test")
+    petab_model = SbmlModel(sbml_model)
+    assert repr(petab_model) == "<SbmlModel 'test'>"