From ad5dda0eebea3e04fdd1ed41eab3b283c54508f0 Mon Sep 17 00:00:00 2001
From: javoha <jan.von.harten@rwth-aachen.de>
Date: Wed, 26 Mar 2025 15:42:32 +0100
Subject: [PATCH] Added test as example for outliers in gempy v3.

---
 test/test_modules/test_outliers.py | 53 ++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 test/test_modules/test_outliers.py

diff --git a/test/test_modules/test_outliers.py b/test/test_modules/test_outliers.py
new file mode 100644
index 00000000..03847f7f
--- /dev/null
+++ b/test/test_modules/test_outliers.py
@@ -0,0 +1,53 @@
+import gempy as gp
+from gempy.optional_dependencies import require_gempy_viewer
+from gempy.core.data.enumerators import ExampleModel
+from gempy.core.data.grid_modules import RegularGrid
+import numpy as np
+
+PLOT = True
+
+
+def test_outliers_model_1():
+    # Path to input data
+    data_path = "https://raw.githubusercontent.com/cgre-aachen/gempy_data/master/"
+    path_to_data = data_path + "/data/input_data/video_tutorials_v3/"
+
+    # Create instance of geomodel
+    model = gp.create_geomodel(
+        project_name='tutorial_model_onlap_1',
+        extent=[0, 2000, 0, 1000, 0, 1000],
+        resolution=[100, 50, 50],
+        importer_helper=gp.data.ImporterHelper(
+            path_to_orientations=path_to_data + "tutorial_model_onlap_1_orientations.csv?cache=",
+            path_to_surface_points=path_to_data + "tutorial_model_onlap_1_surface_points.csv?cache="
+        )
+    )
+
+    # Map geological series to surfaces
+    gp.map_stack_to_surfaces(
+        gempy_model=model,
+        mapping_object={
+            "Young_Series": ("basin_fill_2", "basin_fill_1"),
+            "Old_Series": ("basin_top", "basin_bottom")
+        }
+    )
+
+    # Set the relation of the youngest group to Onlap
+    from gempy_engine.core.data.stack_relation_type import StackRelationType
+    model.structural_frame.structural_groups[0].structural_relation = StackRelationType.ONLAP
+
+    # Compute a solution for the model
+    gp.compute_model(model)
+
+    # Assert
+    arrays = model.solutions.raw_arrays  # * arrays is equivalent to gempy v2 solutions
+    assert arrays.scalar_field_matrix.shape == (2, 250_000)  # * 2 groups, 250000 points
+
+    if PLOT:
+        gpv = require_gempy_viewer()
+        gpv.plot_2d(
+            model=model,
+            show_data=False,
+            show_boundaries=False,
+            show=True
+        )