Skip to content
Next Next commit
[TEST] Adding tests for options preset
  • Loading branch information
Leguark committed May 2, 2025
commit 1ad4a640a9dcd09ae2dc2a56cfe17e0b265fc840
41 changes: 41 additions & 0 deletions test/test_core/test_options_presets.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
import gempy as gp
from gempy.core.data.enumerators import ExampleModel
from gempy.optional_dependencies import require_gempy_viewer

PLOT = True


def test_default_options():
model = gp.generate_example_model(ExampleModel.COMBINATION, compute_model=False)

# TODO: Change options here

gp.compute_model(
gempy_model=model,
engine_config=gp.data.GemPyEngineConfig(
backend=gp.data.AvailableBackends.PYTORCH
)
)

if PLOT:
gpv = require_gempy_viewer()
gpv.plot_3d(model, show_data=True, image=True)


def test_fast_options():
model = gp.generate_example_model(ExampleModel.COMBINATION, compute_model=False)

# TODO: Change options here

gp.compute_model(
gempy_model=model,
engine_config=gp.data.GemPyEngineConfig(
backend=gp.data.AvailableBackends.PYTORCH
)
)

if PLOT:
gpv = require_gempy_viewer()
gpv.plot_3d(model, show_data=True, image=True)