catboost-sys 0.1.6

#![allow(non_upper_case_globals)]
#![allow(non_camel_case_types)]
#![allow(non_snake_case)]

include!(concat!(env!("OUT_DIR"), "/bindings.rs"));

#[cfg(test)]
mod tests {
    use super::*;
    use std::ffi::CStr;

    #[test]
    fn it_works() {
        let model_handle = unsafe { ModelCalcerCreate() };
        let ret_val = unsafe {
            LoadFullModelFromFile(
                model_handle,
                std::ffi::CString::new("files/model.bin").unwrap().as_ptr(),
            )
        };
        if !ret_val {
            let c_str = unsafe { CStr::from_ptr(GetErrorString()) };
            let str_slice = c_str.to_str().unwrap();
            panic!(str_slice);
        }

        let tree_count = unsafe { GetTreeCount(model_handle) };
        assert_eq!(tree_count, 1000);

        let float_features_count = unsafe { GetFloatFeaturesCount(model_handle) };
        assert_eq!(float_features_count, 3);
        let cat_features_count = unsafe { GetCatFeaturesCount(model_handle) };
        assert_eq!(cat_features_count, 1);

        unsafe { ModelCalcerDelete(model_handle) };
    }

    #[test]
    fn it_works_buffer() {
        let buffer = read_fast("files/model.bin").unwrap();
        let model_handle = unsafe { ModelCalcerCreate() };
        let ret_val = unsafe {
            LoadFullModelFromBuffer(
                model_handle,
                buffer.as_ptr() as *const std::os::raw::c_void,
                buffer.len(),
            )
        };
        if !ret_val {
            let c_str = unsafe { CStr::from_ptr(GetErrorString()) };
            let str_slice = c_str.to_str().unwrap();
            panic!(str_slice);
        }

        let tree_count = unsafe { GetTreeCount(model_handle) };
        assert_eq!(tree_count, 1000);

        let float_features_count = unsafe { GetFloatFeaturesCount(model_handle) };
        assert_eq!(float_features_count, 3);
        let cat_features_count = unsafe { GetCatFeaturesCount(model_handle) };
        assert_eq!(cat_features_count, 1);

        unsafe { ModelCalcerDelete(model_handle) };
    }
    use std::io::Read;
    fn read_fast<P: AsRef<std::path::Path>>(path: P) -> std::io::Result<Vec<u8>> {
        let mut file = std::fs::File::open(path)?;
        let meta = file.metadata()?;
        let size = meta.len() as usize;
        let mut data = Vec::with_capacity(size);
        data.resize(size, 0);
        file.read_exact(&mut data)?;
        Ok(data)
    }
}