mirror of
https://github.com/simon987/Simple-Incremental-Search-Tool.git
synced 2025-04-17 09:16:44 +00:00
58 lines
2.3 KiB
Python
58 lines
2.3 KiB
Python
from unittest import TestCase
|
|
from thumbnail import ThumbnailGenerator
|
|
from PIL import Image
|
|
import os
|
|
import shutil
|
|
|
|
|
|
class ThumbnailGeneratorTest(TestCase):
|
|
|
|
def test_generate(self):
|
|
|
|
generator = ThumbnailGenerator(300)
|
|
# Original image is 420x315
|
|
generator.generate("test_folder/sample_1.jpg", "test_thumb1.jpg")
|
|
|
|
img = Image.open("test_thumb1.jpg")
|
|
width, height = img.size
|
|
img.close()
|
|
|
|
self.assertEqual(300, width)
|
|
self.assertEqual(225, height)
|
|
|
|
if os.path.isfile("test_thumb1.jpg"):
|
|
os.remove("test_thumb1.jpg")
|
|
|
|
def test_generate_all(self):
|
|
shutil.rmtree("test_thumbnails")
|
|
|
|
generator = ThumbnailGenerator(300)
|
|
|
|
docs = [{'_source': {'path': 'test_folder', 'name': 'books.csv'}, '_id': 'books.csv-ID'},
|
|
{'_source': {'path': 'test_folder', 'name': 'sample_3.jpg'}, '_id': 'sample_3.jpg-ID'},
|
|
{'_source': {'path': 'test_folder', 'name': 'sample_5.png'}, '_id': 'sample_5.png-ID'},
|
|
{'_source': {'path': 'test_folder', 'name': 'sample_6.gif'}, '_id': 'sample_6.gif-ID'},
|
|
{'_source': {'path': 'test_folder', 'name': 'sample_7.bmp'}, '_id': 'sample_7.bmp-ID'},
|
|
{'_source': {'path': 'test_folder', 'name': 'sample_2.jpeg'}, '_id': 'sample_2.jpeg-ID'}]
|
|
|
|
generator.generate_all(docs, "test_thumbnails")
|
|
|
|
self.assertFalse(os.path.isfile("test_thumbnails/books.csv-ID") and
|
|
os.path.getsize("test_thumbnails/books.csv-ID") > 0)
|
|
self.assertTrue(os.path.isfile("test_thumbnails/sample_3.jpg-ID") and
|
|
os.path.getsize("test_thumbnails/sample_3.jpg-ID") > 0)
|
|
self.assertTrue(os.path.isfile("test_thumbnails/sample_2.jpeg-ID") and
|
|
os.path.getsize("test_thumbnails/sample_2.jpeg-ID") > 0)
|
|
self.assertTrue(os.path.isfile("test_thumbnails/sample_5.png-ID") and
|
|
os.path.getsize("test_thumbnails/sample_5.png-ID") > 0)
|
|
self.assertTrue(os.path.isfile("test_thumbnails/sample_6.gif-ID") and
|
|
os.path.getsize("test_thumbnails/sample_6.gif-ID") > 0)
|
|
self.assertTrue(os.path.isfile("test_thumbnails/sample_7.bmp-ID") and
|
|
os.path.getsize("test_thumbnails/sample_7.bmp-ID") > 0)
|
|
|
|
|
|
|
|
|
|
|
|
|