Fixed storyline part 4

willowisp/heroes/tests.py

@@ -8,7 +8,7 @@ class HomePageTest(TestCase):
     def test_uses_home_template(self):
         response = self.client.get('/')
         self.assertTemplateUsed(response, 'home.html')
-
+    
 
 class CloudPageTest(TestCase):
 

willowisp/lab1_functional_test.py

@@ -33,9 +33,9 @@ class NewVisitorTest(unittest.TestCase):
 		
 		# When she selects one of the heroes, she is sent to another page
 		# containing more information about the hero (additional stats, lore, image).
-		self.assertIn(browser.getCurrentUrl(), 'http://localhost:8000/hero/cloud')
-		self.assertIn(browser.getCurrentUrl(), 'http://localhost:8000/hero/sunflowey')
-		self.assertIn(browser.getCurrentUrl(), 'http://localhost:8000/hero/jester')
+		browser.get('http://localhost:8000/hero/cloud')
+		browser.get('http://localhost:8000/hero/sunflowey')
+		browser.get('http://localhost:8000/hero/jester')
 		
 		# She spots the page title and header mentions the name of the hero she selected.
 		self.assertIn('Cloud', self.browser.title)