def test_section_dictionaries_cross_section():
    geo_model = create_example_model('Horizontal_layers')
    module = GemPyModule(geo_model=geo_model,
                         extent=extent,
                         box=[1000, 800],
                         load_examples=False)
    module.setup(frame)
    module.set_section_dict((10, 10), (500, 500), "Section1")
    module.set_section_dict((100, 100), (500, 10), "Section2")
    module.show_actual_model()

    assert module.section_dict == {
        'Model: Horizontal_layers': ([0.0, 500.0], [1000.0, 500.0], [150,
                                                                     100]),
        'Section1': ([10, 10], [500, 500], [150, 100]),
        'Section2': ([100, 100], [500, 10], [150, 100]),
    }

    module.remove_section_dict("Section2")

    assert module.section_dict == {
        'Model: Horizontal_layers': ([0.0, 500.0], [1000.0, 500.0], [150,
                                                                     100]),
        'Section1': ([10, 10], [500, 500], [150, 100])
    }
def test_plot_cross_sections():
    geo_model = create_example_model('Horizontal_layers')
    module = GemPyModule(geo_model=geo_model,
                         extent=extent,
                         box=[1000, 800],
                         load_examples=False)
    sb_params = module.update(pytest.sb_params)
    sb_params['fig'].show()
    module.set_section_dict((10, 10), (500, 500), "Section1")
    module.set_section_dict((100, 100), (500, 10), "Section2")
    module.show_actual_model()
    module.show_section_traces()
    module.show_cross_section("Section1")
    module.show_cross_section("Section2")
    module.show_geological_map()