diff --git a/test/test_embedart.py b/test/test_embedart.py index 4a4f157ae..729f5853d 100644 --- a/test/test_embedart.py +++ b/test/test_embedart.py @@ -124,11 +124,8 @@ class EmbedartCliTest(_common.TestCase, TestHelper): self.run_command('extractart', '-n', 'extracted') - path = os.path.join(albumpath.decode('utf-8'), 'extracted.') - found = any(map(lambda ext: os.path.exists(path + ext), - ['png', 'jpeg', 'jpg'])) - - self.assertTrue(found, 'Extracted image was not found!') + self.assertExists(os.path.join(albumpath.decode('utf-8'), + 'extracted.png')) class EmbedartTest(unittest.TestCase):