Esempio n. 1
0
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]
Esempio n. 2
0
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()