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()