diff --git a/website/builder/test_site_assets.py b/website/builder/test_site_assets.py index cf6fb7f9..4ad405d8 100644 --- a/website/builder/test_site_assets.py +++ b/website/builder/test_site_assets.py @@ -6,7 +6,7 @@ from collections import Counter from pathlib import Path sys.path.insert(0, str(Path(__file__).resolve().parent)) -from site_assets import heading_slug, optimize_images, referenced_paths, search_document +from site_assets import add_image_sizes, heading_slug, optimize_images, referenced_paths, search_document from build import github_slug, hot_spans, strip_hot @@ -68,5 +68,21 @@ class SearchIndexTests(unittest.TestCase): self.assertEqual(outputs[0], outputs[1]) + def test_image_sizes_keep_aspect_ratio(self): + from PIL import Image + with tempfile.TemporaryDirectory() as directory: + dist = Path(directory) + (dist / 'pics').mkdir() + Image.new('RGB', (1520, 600)).save(dist / 'pics' / 'a.png') + (dist / 'a.md').write_text('\n\n' + '\n\n' + '```html\n\n```\n') + self.assertEqual(add_image_sizes(dist), 2) + self.assertEqual((dist / 'a.md').read_text(), + '\n\n' + '\n\n' + '```html\n\n```\n') + + if __name__ == '__main__': unittest.main()