Support single entity in write

src/simapy/sima_writer.py

@@ -15,20 +15,23 @@
 class SIMAWriter():
     """ Export entites as SIMA objects"""
 
-    def write(self, models: Sequence[MOAO], filename: str, indent=0):
+    def write(self, models: Sequence[MOAO] | MOAO, filename: str, indent=0):
         """Write SIMA models to file"""
         content = self.__to_model_content(models)
         DMTWriter().write(content, filename, indent=indent)
 
-    def __to_model_content(self, models: Sequence[Entity]) -> ModelContent:
+    def __to_model_content(self, models: Sequence[Entity] | Entity) -> ModelContent:
         content = ModelContent()
         header = Header()
 
         for name,version in packages.items():
             header.packages.append(PackageInfo(name=name,version=version))
 
         content.header = header
-        content.contents.extend(models)
+        if isinstance(models, Entity):
+            content.contents.append(models)
+        else:
+            content.contents.extend(models)
         return content
 
     def to_dict(self, models: Sequence[MOAO]) -> Dict: