sorer 0.1.0

use sorer::dataframe::*;
use sorer::schema::*;

use std::fs::File;
use std::io::BufReader;
use std::path::Path;

#[test]
fn get_col_type() {
    let col_type_tests = vec![
        ("tests/0.sor", 0, DataType::Bool),
        ("tests/1.sor", 0, DataType::String),
        ("tests/2.sor", 0, DataType::Bool),
        ("tests/2.sor", 1, DataType::Int),
        ("tests/2.sor", 2, DataType::Float),
        ("tests/2.sor", 3, DataType::String),
        ("tests/3.sor", 4, DataType::Bool),
    ];

    for t in col_type_tests {
        let f = File::open(Path::new(t.0)).unwrap();
        let reader = BufReader::new(f);
        let s = infer_schema(reader);

        assert_eq!(*s.get(t.1).unwrap(), t.2);
    }
}

#[test]
fn is_missing_idx() {
    let is_missing_tests = vec![
        (String::from("tests/0.sor"), 0, 0, true),
        (String::from("tests/1.sor"), 0, 1, false),
        (String::from("tests/2.sor"), 1, 0, true),
        (String::from("tests/2.sor"), 1, 1, false),
    ];

    for t in is_missing_tests {
        let data_frame: DataFrame = DataFrame::from_file(t.0, 0, std::u64::MAX);

        assert_eq!(data_frame.get(t.1, t.2) == Data::Null, t.3);
    }

    // special case
    // ./sorer./sorer -f 1.sor -from 1 -len 74 -is_missing_idx 0 0
    let data_frame: DataFrame =
        DataFrame::from_file(String::from("tests/1.sor"), 1, 74);

    assert_eq!(data_frame.get(0, 0) == Data::Null, false);
}

// NOTE: This test is ignored by default since running `cargo test` uses the debug build, which is
// much much slower than the release version (release is roughly 60x faster).
// If you want to run this test, run `cargo test --release -- --ignored`
#[test]
#[ignore]
fn print_col_idx() {
    let print_col_idx_tests = vec![
        (
            String::from("tests/1.sor"),
            0,
            3,
            Data::String("+1".to_string()),
        ),
        (
            String::from("tests/2.sor"),
            3,
            0,
            Data::String("hi".to_string()),
        ),
        (
            String::from("tests/2.sor"),
            3,
            1,
            Data::String("ho ho ho".to_string()),
        ),
        (String::from("tests/2.sor"), 2, 0, Data::Float(1.2)),
        (String::from("tests/2.sor"), 2, 1, Data::Float(-0.2)),
        (String::from("tests/3.sor"), 2, 10, Data::Float(1.2)),
        (String::from("tests/3.sor"), 2, 384200, Data::Float(1.2)),
        (String::from("tests/4.sor"), 0, 1, Data::Int(2147483647)),
        (String::from("tests/4.sor"), 0, 2, Data::Int(-2147483648)),
        (String::from("tests/4.sor"), 1, 1, Data::Float(-2e-09)),
        (String::from("tests/4.sor"), 1, 2, Data::Float(1e+10)),
    ];

    for t in print_col_idx_tests {
        let data_frame: DataFrame = DataFrame::from_file(t.0, 0, std::u64::MAX);

        assert_eq!(data_frame.get(t.1, t.2), t.3);
    }
    // special case:
    // ./sorer./sorer -f 1.sor -from 1 -len 74 -print_col_idx 0 6
    // "+2.2"
}