From d8e8ba51c1de2904861a6d176e40e82bd53ffefd Mon Sep 17 00:00:00 2001 From: CyC2018 Date: Sun, 1 Nov 2026 20:08:24 +0800 Subject: [PATCH] Update test_site_assets.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 补充图片补高度的测试。 --- website/builder/test_site_assets.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) 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()