def test_no_countries_have_plots_at_first(self):
# Set up
app = Labyrinth(1, 1)
# Invoke
plot_countries = app.get_plot_countries()
# Check
self.assertEqual([], plot_countries)
def test_two_countries_have_plots(self):
# Set up
app = Labyrinth(1, 1)
actual_plot_countries = ["China", "Italy"]
for country_name in actual_plot_countries:
app.get_country(country_name).plots = 2
# Invoke
plot_countries = [c.name for c in app.get_plot_countries()]
# Check
plot_countries.sort()
self.assertEqual(actual_plot_countries, plot_countries)