Esempio n. 1
0
    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)
Esempio n. 2
0
    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)