def test_make_compare_plot_hungary(): with_local = c.choose_random_counties(exclude_region="Baranya", size=18) axes, cases, deaths = c.make_compare_plot_hungary("Baranya", compare_with_local=with_local) assert deaths is None assert type(cases) == pd.core.frame.DataFrame assert cases.shape[1] == 20 # counties and the capital city
def test_choose_random_counties(): # Hungary related with_local = c.choose_random_counties(exclude_region="Baranya", size=18) # print(with_local) assert 'Baranya' not in with_local assert len(with_local) == 19