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)