[ENH] Add example generator and initial speed test

Introduce `examples_generator.py` to create sample geological models, including a "two wells" example. Add `test_speed_I` for benchmarking model computation and validation of interpolation options.

Author: Leguark

docs/dev_logs/2025_May.md

@@ -5,7 +5,8 @@
   - [ ] Keep extracting code into modules and functions 
   - [ ] Posterior analysis for gempy 
 - [ ] Investigate speed
-- [ ] Investigate segmentation function 
+- [ ] Investigate segmentation function
+  https://chatgpt.com/share/68172f53-4f7c-8000-8f42-205a033ed899
 - [ ] Create a robust api to make sure we do not mess up assigning priors to gempy parameters
 - [ ] do we need to use index_put or just normal indexing?
 --- Vector examples
@@ -18,9 +19,13 @@
 ### Doing:
 - [x] Compressing code
 - [x] GemPy posterior visualization
-- [ ] Add proper thickness likelihood
+- [x] Add proper thickness likelihood
 - [ ] Speed
-
+  - [ ] Time test for torch CPU
+  - [ ] Time test for keops CPU
+  - [ ] Time test for keops GPU
+  - [ ] Options and config presets
+   
 ### Possible Design:
 - Geological Model
 - Priors

gempy_probability/plugins/examples_generator.py

@@ -0,0 +1,41 @@
+from enum import Enum, auto
+import gempy as gp
+import os
+from gempy_engine.core.backend_tensor import BackendTensor
+
+
+class ExampleProbModel(Enum):
+    TWO_WELLS = auto()
+
+
+def generate_example_model(example_model: ExampleProbModel, compute_model: bool = True) -> gp.data.GeoModel:
+    match example_model:
+        case ExampleProbModel.TWO_WELLS:
+            return _generate_two_wells_model(compute_model)
+        case _:
+            raise ValueError(f"Example model {example_model} not found.")
+
+
+def _generate_two_wells_model(compute_model: bool) -> gp.data.GeoModel:
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    data_path = os.path.abspath(os.path.join(current_dir, '..', '..', 'examples', 'tutorials', 'data'))
+    geo_model = gp.create_geomodel(
+        project_name='Wells',
+        extent=[0, 12000, -500, 500, 0, 4000],
+        refinement=4,
+        importer_helper=gp.data.ImporterHelper(
+            path_to_orientations=os.path.join(data_path, "2-layers", "2-layers_orientations.csv"),
+            path_to_surface_points=os.path.join(data_path, "2-layers", "2-layers_surface_points.csv")
+        )
+    )
+
+    BackendTensor.change_backend_gempy(engine_backend=gp.data.AvailableBackends.PYTORCH)
+
+    geo_model.interpolation_options.uni_degree = 0
+    geo_model.interpolation_options.mesh_extraction = False
+
+    if compute_model:
+        # Compute the geological model
+        gp.compute_model(geo_model)
+
+    return geo_model

tests/test_benchmark/test_speed_I.py

@@ -0,0 +1,17 @@
+import gempy as gp
+import gempy_viewer as gpv
+from gempy_probability.plugins.examples_generator import generate_example_model, ExampleProbModel
+
+
+def test_speed_I():
+    two_wells: gp.data.GeoModel = generate_example_model(
+        example_model=ExampleProbModel.TWO_WELLS,
+        compute_model=False
+    )
+
+    assert two_wells.interpolation_options.number_octree_levels == 4, "Number of octrees should be 4"
+
+    gp.compute_model(two_wells)
+
+    if PLOT := False:
+        gpv.plot_2d(two_wells, show_scalar=False)