File size: 1,139 Bytes
ba54ea9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | from datetime import datetime
from PIL import Image
from recap.ingestion.image import load_image_event
def test_loads_image_with_provided_date_and_category(tmp_path):
p = tmp_path / "fundus.png"
Image.new("RGB", (100, 100), "white").save(p)
e = load_image_event(
str(p),
category="scan",
title="Right fundus",
date_iso="2023-04-01",
source_id="fundus_2023.png",
)
assert e.category == "scan"
assert e.date == datetime.fromisoformat("2023-04-01T00:00:00+00:00")
assert e.source == "fundus_2023.png"
assert "fundus" in e.title.lower()
def test_default_source_id_is_filename(tmp_path):
p = tmp_path / "ct_chest.png"
Image.new("RGB", (10, 10), "black").save(p)
e = load_image_event(str(p), category="scan", title="CT chest", date_iso="2024-01-15")
assert e.source == "ct_chest.png"
def test_image_path_preserved_in_metadata(tmp_path):
p = tmp_path / "wound.jpg"
Image.new("RGB", (10, 10), "red").save(p)
e = load_image_event(str(p), category="photo", title="Wound day 7", date_iso="2024-06-20")
assert e.metadata["path"] == str(p)
|