def test_plot_mort_rates(): fig = parameter_plots.plot_mort_rates( base_params, include_title=True) assert fig
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)
def test_plot_mort_rates(): fig = parameter_plots.plot_mort_rates(base_params) assert fig