Fix test: now returns a png

2023-06-03 17:55:00 +03:00 · 2023-06-03 17:55:00 +03:00 · 1c669d8bb0
parent 0f59831dc2
commit 1c669d8bb0
1 changed files with 2 additions and 2 deletions
--- a/tests/test_album.py
+++ b/tests/test_album.py
@ -94,8 +94,8 @@ class TestPyLastAlbum(TestPyLastWithLastFm):
        image = album.get_cover_image()

        # Assert
-        self.assert_startswith(image, "https://")
-        self.assert_endswith(image, ".gif")
+        assert image.startswith("https://")
+        assert image.endswith(".gif") or image.endswith(".png")

    def test_mbid(self) -> None:
        # Arrange