File size: 363 Bytes
717c92d
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from pathlib import Path
from imgshape.report import generate_markdown_report
def test_generate_markdown(tmp_path):
    md = tmp_path / "report.md"
    stats = {"entropy_mean":4.0}
    generate_markdown_report(md, stats, {}, {}, {"order":[], "augmentations":[]})
    assert md.exists()
    content = md.read_text()
    assert "imgshape dataset report" in content