diff --git a/tests/test_imgurfetcher.py b/tests/test_imgurfetcher.py index cc17a62..eeac25c 100644 --- a/tests/test_imgurfetcher.py +++ b/tests/test_imgurfetcher.py @@ -127,10 +127,10 @@ def test_select_image(self): """ def test_get_image_from_album(self): - with self.assertRaises(AssertionError): + with self.assertRaises(ValueError): self.fetcher._get_image_from_album(None) - with self.assertRaises(AssertionError): + with self.assertRaises(ValueError): self.fetcher._get_image_from_album(self.gallery[0]) with patch("bg_daemon.fetchers.imgurfetcher.ImgurClient") as \