def test_layout_returns_blocks(layout_predictor, test_image): layout_results = layout_predictor([test_image]) assert len(layout_results) == 1 res = layout_results[0] assert res.image_bbox == [0, 0, 1024, 1024] if res.error: # Server may not be running in CI environments without llama-server return assert isinstance(res.bboxes, list) for box in res.bboxes: assert box.label assert box.count >= 0 assert isinstance(box.position, int)