Exemple #1
0
 def test_command_save(self):
     xkcd.sel_comic = 1000
     xkcd.command_save("1000.png")
     with open("1000.png", "rb") as saved_image:
         content = saved_image.read()
     try:
         with open("1000_correct.png", "rb") as correct_image:
             correct_content = correct_image.read()
     except IOError:
         with open("tests/1000_correct.png", "rb") as correct_image:
             correct_content = correct_image.read()
     self.assertEqual(content, correct_content)
     shutil.rmtree(xkcd.tmpimg_location)
     os.remove("1000.png")
Exemple #2
0
 def test_command_save_404(self):
     xkcd.sel_comic = 1608
     output = xkcd.command_save()
     excepted_output = "No image for comic found (maybe it's interactive?)"
     self.assertEqual(output, excepted_output)