예제 #1
0
def test_plot_mort_rates():
    fig = parameter_plots.plot_mort_rates(
        base_params, include_title=True)
    assert fig
예제 #2
0
def test_plot_mort_rates_save_fig(tmpdir):
    parameter_plots.plot_mort_rates(
            base_params, path=tmpdir)
    img = mpimg.imread(os.path.join(tmpdir, 'mortality_rates.png'))

    assert isinstance(img, np.ndarray)
예제 #3
0
def test_plot_mort_rates():
    fig = parameter_plots.plot_mort_rates(base_params)
    assert fig