Skip to content

Commit

Permalink
Update screenshot test to use updated b64decode method
Browse files Browse the repository at this point in the history
  • Loading branch information
AutomatedTester committed Jan 18, 2016
1 parent c1154cc commit 715148b
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ class ScreenshotTests(unittest.TestCase):

def test_get_screenshot_as_base64(self):
self._loadSimplePage()
result = base64.decodestring(self.driver.get_screenshot_as_base64())
result = base64.b64decode(self.driver.get_screenshot_as_base64())
self.assertEqual(imghdr.what('', result), 'png')

def test_get_screenshot_as_png(self):
Expand Down

0 comments on commit 715148b

Please sign in to comment.