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 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()