exodata-core 0.2.0

Core data loading, metadata, insight, and table logic for Exoplanets Catalog
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use exo_core::tables::common::{
    count_non_null_values, create_histogram, get_numeric_stats, load_parquet,
};

type TestResult = Result<(), Box<dyn std::error::Error>>;

const EXOPLANETS_FIXTURE: &str = "tests/fixtures/exoplanets.fixture";

#[test]
fn test_load_parquet() -> TestResult {
    let df = load_parquet(EXOPLANETS_FIXTURE, None)?;
    assert_eq!(df.height(), 100, "Should have 100 sampled rows");
    assert!(df.width() > 0, "Should have columns");
    Ok(())
}

#[test]
fn test_load_parquet_with_limit() -> TestResult {
    let df = load_parquet(EXOPLANETS_FIXTURE, Some(10))?;
    assert_eq!(df.height(), 10, "Should limit to 10 rows");
    Ok(())
}

#[test]
fn test_count_non_null_values() -> TestResult {
    let df = load_parquet(EXOPLANETS_FIXTURE, None)?;
    let count = count_non_null_values(&df, "pl_name")?;
    assert!(count > 0, "Should have non-null planet names");
    assert!(count <= 100, "Count should not exceed total rows");
    Ok(())
}
#[test]
fn test_get_numeric_stats() -> TestResult {
    let df = load_parquet(EXOPLANETS_FIXTURE, None)?;

    if let Some(stats) = get_numeric_stats(&df, "pl_masse")? {
        assert!(stats.count > 0);
        assert!(stats.min <= stats.max);
        assert!(stats.mean >= stats.min);
        assert!(stats.mean <= stats.max);
        assert!(stats.std >= 0.0);
    }
    Ok(())
}

#[test]
fn test_create_histogram() -> TestResult {
    let df = load_parquet(EXOPLANETS_FIXTURE, None)?;
    let histogram = create_histogram(&df, "pl_orbper", 0.0, 1000.0, 10)?;

    assert_eq!(histogram.len(), 10, "Should have 10 bins");

    // Verify bin structure
    for bin in &histogram {
        assert!(bin.min < bin.max, "Bin min should be less than max");
    }
    Ok(())
}