Spaces:
Running
Running
| from __future__ import annotations | |
| import hashlib | |
| import pytest | |
| from app.services.figure_export import export_figure | |
| SVG = """<svg xmlns='http://www.w3.org/2000/svg' width='120' height='80' viewBox='0 0 120 80'> | |
| <rect x='0' y='0' width='120' height='80' fill='white'/> | |
| <text x='10' y='30'>A</text><line x1='10' y1='50' x2='110' y2='50' stroke='black'/> | |
| </svg>""" | |
| def test_svg_is_canonical_and_content_addressed(): | |
| artifact = export_figure(SVG, "svg") | |
| assert artifact.content_type == "image/svg+xml" | |
| assert artifact.payload == SVG.encode() | |
| assert artifact.sha256 == hashlib.sha256(artifact.payload).hexdigest() | |
| assert artifact.dpi is None | |
| def test_invalid_raster_dpi_is_rejected(): | |
| with pytest.raises(ValueError): | |
| export_figure(SVG, "png", 299) | |
| with pytest.raises(ValueError): | |
| export_figure(SVG, "tiff", 601) | |
| def test_png_pdf_tiff_are_nonempty_real_exports_when_renderer_available(): | |
| pytest.importorskip("cairosvg") | |
| pytest.importorskip("PIL") | |
| png = export_figure(SVG, "png", 300) | |
| assert png.content_type == "image/png" | |
| assert len(png.payload) > 100 | |
| assert png.dpi == 300 | |
| assert png.width_px and png.width_px >= 300 | |
| pdf = export_figure(SVG, "pdf", 300) | |
| assert pdf.content_type == "application/pdf" | |
| assert len(pdf.payload) > 100 | |
| assert pdf.dpi is None | |
| tiff = export_figure(SVG, "tiff", 600) | |
| assert tiff.content_type == "image/tiff" | |
| assert len(tiff.payload) > 100 | |
| assert tiff.dpi == 600 | |
| assert tiff.sha256 == hashlib.sha256(tiff.payload).hexdigest() | |