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