use std::path::PathBuf;

use cocotools::COCO;

#[test]
#[allow(clippy::unwrap_used, clippy::float_cmp)]
fn load_from_file() {
    let annotations_file_path = PathBuf::from("../data_samples/coco_25k/annotations.json");
    let image_folder_path = PathBuf::from("../data_samples/coco_25k/images");
    let dataset = COCO::new(&annotations_file_path, &image_folder_path).unwrap();

    assert_eq!(
        dataset.get_img(17627).unwrap().file_name,
        "000000017627.jpg"
    );
    assert_eq!(dataset.get_ann(128_189).unwrap().area, 71436.89385);
    assert_eq!(dataset.get_cat(86).unwrap().name, "vase");
}