import reverse

game/end_to_end_tests/test_level_selection.py

@@ -2,6 +2,7 @@
 from common.tests.utils.organisation import create_organisation_directly
 from common.tests.utils.student import create_school_student_directly
 from common.tests.utils.teacher import signup_teacher_directly
+from django.urls import reverse
 from hamcrest import assert_that, ends_with, equal_to
 from selenium.common.exceptions import NoSuchElementException
 from selenium.webdriver.common.by import By
@@ -69,7 +70,7 @@ def test_redirect_to_levelless_episode(self):
         expected_url = levels_page.browser.current_url
 
         page = self._go_to_path(reverse("play_python_default_level", kwargs={"level_name": "41"}))
-        
+
         next_button = self.selenium.find_element(By.ID, "next_button")
         assert WebDriverWait(self.selenium, 10).until(
             EC.element_to_be_clickable(next_button)