22 lines
517 B
Python
22 lines
517 B
Python
|
import unittest
|
||
|
|
import os
|
||
|
|
|
||
|
|
from imgin import get
|
||
|
|
|
||
|
|
CACHE_DIR = '/tmp/imgin-imgur-images-single/'
|
||
|
|
|
||
|
|
class TestSingleImage(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_single_image(self):
|
||
|
|
img = "7TiLluI.jpg"
|
||
|
|
get(f"https://imgur.com/{img}", CACHE_DIR)
|
||
|
|
with open(f"tests/test_images/{img}", "rb") as expected:
|
||
|
|
with open(CACHE_DIR + img, "rb") as actual:
|
||
|
|
self.assertEqual(actual.read(), expected.read())
|
||
|
|
|
||
|
|
try:
|
||
|
|
os.mkdir(CACHE_DIR)
|
||
|
|
except FileExistsError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
unittest.main()
|