Skip to content

Commit 2db1530

Browse files
author
wordofglass
committed
fetchart: flake8 fixes
1 parent 5d00ca8 commit 2db1530

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

test/test_art.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,22 +46,24 @@ def setUp(self):
4646

4747

4848
class FetchImageHelper(_common.TestCase):
49-
"""Helper mixin for mocking requests when fetching images
49+
"""Helper mixin for mocking requests when fetching images
5050
with remote art sources.
5151
"""
5252
@responses.activate
5353
def run(self, *args, **kwargs):
5454
super(FetchImageHelper, self).run(*args, **kwargs)
5555

56+
IMAGEHEADER = {'image/jpeg': b'\x00' * 6 + b'JFIF',
57+
'image/png': b'\211PNG\r\n\032\n', }
58+
5659
def mock_response(self, url, content_type='image/jpeg', file_type=None):
57-
IMAGEHEADER = {'image/jpeg': b'\x00' * 6 + b'JFIF',
58-
'image/png': b'\211PNG\r\n\032\n', }
5960
if file_type is None:
6061
file_type = content_type
6162
responses.add(responses.GET, url,
6263
content_type=content_type,
6364
# imghdr reads 32 bytes
64-
body=IMAGEHEADER.get(file_type, b'').ljust(32, b'\x00'))
65+
body=self.IMAGEHEADER.get(
66+
file_type, b'').ljust(32, b'\x00'))
6567

6668

6769
class FetchImageTest(FetchImageHelper, UseThePlugin):

0 commit comments

Comments
 (0)