def test_load_release_area():
module = LandslideSimulation(extent=extent)
module.Load_Area.loadTopo(test_data['landslide_topo'] + 'Topography_3.npz')
assert module.Load_Area.file_id == '3'
module.load_release_area(test_data['landslide_release'])
lst = ['ReleaseArea_3_1.npy', 'ReleaseArea_3_2.npy', 'ReleaseArea_3_2.npy']
assert [i in lst for i in module.release_options]
lst2 = ['1', '2', '3']
assert [i in lst for i in module.release_id_all]
def load_modules(self, gempy_module: bool, **kwargs):
from sandbox.modules import TopoModule, GradientModule, LoadSaveTopoModule, LandslideSimulation, SearchMethodsModule
from sandbox.projector import ContourLinesModule, CmapModule
self.Modules = {}
self.Modules['ContourLinesModule'] = ContourLinesModule(
extent=self.sensor.extent)
self.Modules['CmapModule'] = CmapModule(extent=self.sensor.extent)
self.Modules['TopoModule'] = TopoModule(extent=self.sensor.extent)
self.Modules['GradientModule'] = GradientModule(
extent=self.sensor.extent)
self.Modules['LoadSaveTopoModule'] = LoadSaveTopoModule(
extent=self.sensor.extent)
self.Modules['LandslideSimulation'] = LandslideSimulation(
extent=self.sensor.extent)
self.Modules['SearchMethodsModule'] = SearchMethodsModule(
extent=self.sensor.extent)
self.Modules['SearchMethodsModule'].update_mesh(
self.sensor.get_frame(),
margins_crop=self.Modules['SearchMethodsModule'].margins_crop,
fill_value=0)
self.Modules['SearchMethodsModule'].activate_frame_capture = False
if gempy_module:
from sandbox.modules import GemPyModule
geo_model = kwargs.get('geo_model')
#load_example = kwargs.get('load_examples')
#name_example = kwargs.get('name_example')
self.Modules['GemPyModule'] = GemPyModule(
geo_model=geo_model,
extent=self.sensor.extent,
box=self.sensor.physical_dimensions,
#load_examples=load_example,
#name_example=name_example,
**kwargs)
def test_panel_plot():
module = LandslideSimulation(extent=extent)
module.load_simulation_data_npz(test_data['landslide_simulation'] +
'Sim_Topo1_Rel13_results4sandbox.npz')
module.frame_selector = 10
module.plot_frame_panel()
module.plot_flow_frame.show()
def test_show_widgets():
module = LandslideSimulation(extent=extent)
module.load_simulation_data_npz(test_data['landslide_simulation'] +
'Sim_Topo1_Rel13_results4sandbox.npz')
module.flow_selector = "Velocity"
module.frame_selector = 10
landslide = module.show_widgets()
landslide.show()
def test_show_box_release():
module = LandslideSimulation(extent=extent)
module.Load_Area.loadTopo(test_data['landslide_topo'] + 'Topography_3.npz')
module.load_release_area(test_data['landslide_release'])
module.modify_to_box_coordinates(id='1')
fig, ax = plt.subplots()
ax.imshow(frame,
vmin=extent[-2],
vmax=extent[-1],
cmap='gist_earth_r',
origin='lower')
module.show_box_release(ax, module.release_area)
#TODO assert np.allclose(np.asarray([[74., 72.], [74., 84.],[86., 84.],[86., 72.]]), module.release_area)
fig.show()
def load_modules(self,
gempy_module: bool = False,
gimli_module: bool = False,
torch_module: bool = False,
devito_module: bool = False,
kwargs_gempy_module: dict = {},
):
from sandbox.modules import (TopoModule, GradientModule, LoadSaveTopoModule, LandslideSimulation,
SearchMethodsModule)
from sandbox.projector import ContourLinesModule, CmapModule
self.Modules = {'ContourLinesModule': ContourLinesModule(extent=self.sensor.extent),
'CmapModule': CmapModule(extent=self.sensor.extent),
'TopoModule': TopoModule(extent=self.sensor.extent),
'GradientModule': GradientModule(extent=self.sensor.extent),
'LoadSaveTopoModule': LoadSaveTopoModule(extent=self.sensor.extent),
'LandslideSimulation': LandslideSimulation(extent=self.sensor.extent),
'SearchMethodsModule': SearchMethodsModule(extent=self.sensor.extent)}
#self.Modules['SearchMethodsModule'].update_mesh(self.sensor.get_frame(),
# margins_crop=self.Modules['SearchMethodsModule'].margins_crop,
# fill_value=0)
#self.Modules['SearchMethodsModule'].activate_frame_capture = False
if gempy_module and self._gempy_import:
from sandbox.modules.gempy import GemPyModule
self.Modules['GemPyModule'] = GemPyModule(extent=self.sensor.extent,
box=self.sensor.physical_dimensions,
**kwargs_gempy_module)
if devito_module and self._devito_import:
from sandbox.modules.devito import SeismicModule
self.Modules['SeismicModule'] = SeismicModule(extent=self.sensor.extent)
if gimli_module and self._pygimli_import:
from sandbox.modules.gimli import GeoelectricsModule
self.Modules['GeoelectricsModule'] = GeoelectricsModule(extent=self.sensor.extent)
if torch_module and self._torch_import:
from sandbox.modules.pytorch import LandscapeGeneration
self.Modules['LandscapeGeneration'] = LandscapeGeneration(extent=self.sensor.extent)
def test_plot_landslide():
module = LandslideSimulation(extent=extent)
module.load_simulation_data_npz(test_data['landslide_simulation'] +
'Sim_Topo1_Rel13_results4sandbox.npz')
module.flow_selector = "Velocity"
module.frame_selector = 10
fig, ax = plt.subplots()
ax.imshow(frame,
vmin=extent[-2],
vmax=extent[-1],
cmap='gist_earth_r',
origin='lower')
module.plot_landslide_frame(ax)
fig.show()
module.flow_selector = "Height"
ax.cla()
ax.imshow(frame,
vmin=extent[-2],
vmax=extent[-1],
cmap='gist_earth_r',
origin='lower')
module.plot_landslide_frame(ax)
fig.show()
def test_load_simulation():
module = LandslideSimulation(extent=extent)
module.load_simulation_data_npz(test_data['landslide_simulation'] +
'Sim_Topo1_Rel13_results4sandbox.npz')
assert module.velocity_flow is not None and module.height_flow is not None
def test_update():
module = LandslideSimulation(extent=extent)
update(module)
def test_init():
module = LandslideSimulation(extent=extent)
print(module)