[ENH] Adding logic to test serialization on compute model

Author: Leguark

gempy/API/compute_API.py

@@ -1,4 +1,7 @@
-from typing import Optional
+import dotenv
+import os
+
+from typing import Optional
 
 import numpy as np
 
@@ -14,6 +17,8 @@
 from ..modules.data_manipulation.engine_factory import interpolation_input_from_structural_frame
 from ..optional_dependencies import require_gempy_legacy
 
+dotenv.load_dotenv()
+
 
 def compute_model(gempy_model: GeoModel, engine_config: Optional[GemPyEngineConfig] = None) -> Solutions:
     """
@@ -56,6 +61,12 @@ def compute_model(gempy_model: GeoModel, engine_config: Optional[GemPyEngineConf
         case _:
             raise ValueError(f'Backend {engine_config} not supported')
 
+    if os.getenv("VALIDATE_SERIALIZATION", False):
+        from ..modules.serialization.save_load import save_model
+        import tempfile
+        with tempfile.NamedTemporaryFile(mode='w+', delete=True) as tmp:
+            save_model(model=gempy_model, path=tmp.name, validate_serialization=True)
+
     return gempy_model.solutions
 
 

gempy/modules/serialization/save_load.py

@@ -5,7 +5,7 @@
 import os
 
 
-def save_model(model: GeoModel, path: str, validate_serialization: bool = True):
+def save_model(model: GeoModel, path: str | None = None, validate_serialization: bool = True):
     """
     Save a GeoModel to a file with proper extension validation.
     
@@ -23,6 +23,8 @@ def save_model(model: GeoModel, path: str, validate_serialization: bool = True):
     """
     # Define the valid extension for gempy models
     VALID_EXTENSION = ".gempy"
+    if path is None:
+        path = model.meta.name + VALID_EXTENSION
 
     # Check if path has an extension
     path_obj = pathlib.Path(path)