undo small change, run black

game/end_to_end_tests/test_level_editor.py

@@ -291,8 +291,7 @@ def test_electric_fuel_gauge(self):
         assert "visibility: visible" in electric_fuel_gauge.get_attribute("style")
 
     def test_pigeon(self):
-        """ Test that cows on the map automatically become pigeons when the theme is changed to "city"
-        """
+        """Test that cows on the map automatically become pigeons when the theme is changed to "city" """
         page = self.go_to_level_editor()
         page.go_to_scenery_tab()
 
@@ -302,15 +301,18 @@ def test_pigeon(self):
         )
         ActionChains(self.selenium).drag_and_drop(source_cow, end_space).perform()
 
-        scenery_animal = self.selenium.find_elements(
+        scenery_cow = self.selenium.find_elements(
             By.CSS_SELECTOR, "image[x='0'][y='0']"
         )
-        cow_link = scenery_animal[0].get_attribute("href")
+        cow_link = scenery_cow[0].get_attribute("href")
         assert cow_link == "/static/game/raphael_image/Clarice.svg"
 
         Select(self.selenium.find_element(By.ID, "theme_select")).select_by_value(
             "city"
         )
 
-        pigeon_link = scenery_animal[0].get_attribute("href")
+        scenery_pigeon = self.selenium.find_elements(
+            By.CSS_SELECTOR, "image[x='0'][y='0']"
+        )
+        pigeon_link = scenery_pigeon[0].get_attribute("href")
         assert pigeon_link == "/static/game/raphael_image/pigeon.svg"