| """Tests for ChartQA/A-OKVQA image path resolution.""" | |
| import os | |
| import sys | |
| sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) | |
| from data_utils.paths import CHARTQA_IMAGES_DIR, resolve_image_path | |
| def test_resolve_legacy_chartqa_path(tmp_path, monkeypatch): | |
| import data_utils.paths as paths_mod | |
| img_dir = tmp_path / "data" / "images" / "chartqa" / "images" | |
| img_dir.mkdir(parents=True) | |
| img_file = img_dir / "train_002843.png" | |
| img_file.write_bytes(b"x") | |
| monkeypatch.setattr(paths_mod, "PROJECT_ROOT", str(tmp_path)) | |
| monkeypatch.setattr(paths_mod, "CHARTQA_IMAGES_DIR", str(img_dir)) | |
| monkeypatch.setattr( | |
| paths_mod, | |
| "CHARTQA_DIR", | |
| str(tmp_path / "data" / "images" / "chartqa"), | |
| ) | |
| resolved = paths_mod.resolve_image_path("/chartqa_output/images/train_002843.png") | |
| assert resolved == str(img_file.resolve()) | |
| assert os.path.exists(resolved) | |