def test_modify_surface_values(): mm = gp.DataMutation() mm.add_surfaces(['surface1', 'foo1', 'foo2', 'foo3']) mm.add_surface_points(400, 300, -500, 'foo2') print(mm.surface_points) mm.modify_surface_points(0, Y=800) print(mm.surface_points)
def test_set_is_fault(): mm = gp.DataMutation() mm.add_series(['foo1', 'foo2', 'foo3']) assert (mm.faults.df.index == np.array(['Default series', 'foo1', 'foo2', 'foo3'])).all() assert (mm.faults.faults_relations_df.index == ['Default series', 'foo1', 'foo2', 'foo3']).all() mm.set_is_fault(['foo2']) assert mm.faults.faults_relations_df.loc['foo2', 'foo3'] == True assert mm.faults.faults_relations_df.iloc[2,3] == True mm.set_is_fault(['foo2'], toggle=True)
def test_add_default_orientation(): mm = gp.DataMutation() mm.set_default_surfaces() mm.set_default_orientation()
def test_add_surface_points(): mm = gp.DataMutation() mm.add_surfaces(['surface1', 'foo1', 'foo2', 'foo3']) mm.add_surface_points(400, 300, -500, 'foo2')
def test_modify_order_surfaces(): mm = gp.DataMutation() mm.add_surfaces(['surface1', 'foo1', 'foo2', 'foo3']) mm.modify_order_surfaces(3, 2) assert mm.surfaces.df.iloc[2, 0] == 'foo2'
def test_rename_surface(): mm = gp.DataMutation() mm.add_surfaces(['surface1', 'foo1', 'foo2', 'foo3']) mm.rename_surfaces({'foo1': 'changed'}) assert mm.surfaces.df.loc[1, 'surface'] == 'changed'
# These two lines are necessary only if GemPy is not installed import sys, os sys.path.append("../..") # Importing GemPy import gempy as gp # Importing auxiliary libraries import numpy as np import pandas as pn import matplotlib.pyplot as plt import pytest mm = gp.DataMutation() mm.add_surfaces(['surface1', 'foo1', 'foo2', 'foo3']) def test_add_surface_points_raise_non_surface(): with pytest.raises(ValueError): mm.add_surface_points(400, 300, -500, 'surface5') def test_add_surface(): mm.add_surfaces(['a_foo1', 'a_foo2']) with pytest.raises(ValueError, match=r'.* not include old categories.*'): mm.add_surfaces('a_foo1') mm.add_surfaces('a_foo3')
def test_set_is_fault(): mm = gp.DataMutation() mm.add_series(['foo1', 'foo2', 'foo3']) mm.set_is_fault(['foo2']) mm.set_is_fault(['foo2'], toggle=True)