ragc-core 0.1.1

Core compression and decompression algorithms for the AGC genome compression format
Documentation 
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
59
60
61
62
63
64
65
66
67
68
69

#![allow(clippy::all)]
// Integration test for StreamingQueueCompressor
// Verifies basic push() API and worker thread operation

use ahash::AHashSet;
use ragc_core::{StreamingQueueCompressor, StreamingQueueConfig};

#[test]
fn test_streaming_queue_basic_flow() {
    // Create a small test dataset
    let test_data = vec![
        ("sample1".to_string(), "chr1".to_string(), vec![0u8; 1000]), // A x 1000
        ("sample1".to_string(), "chr2".to_string(), vec![1u8; 1000]), // C x 1000
        ("sample2".to_string(), "chr1".to_string(), vec![2u8; 1000]), // G x 1000
    ];

    // Create compressor with small queue for testing
    let config = StreamingQueueConfig {
        queue_capacity: 10 * 1024 * 1024, // 10 MB for test
        num_threads: 2,
        verbosity: 0, // Quiet for tests
        ..Default::default()
    };

    let splitters = AHashSet::new(); // Will be determined from first contig
    let mut compressor = StreamingQueueCompressor::with_splitters(
        "/tmp/test_streaming_queue.agc",
        config,
        splitters,
    )
    .expect("Failed to create compressor");

    // Push all contigs
    for (sample, contig, data) in test_data {
        compressor
            .push(sample, contig, data)
            .expect("Failed to push contig");
    }

    // Finalize (waits for workers)
    compressor.finalize().expect("Failed to finalize");

    // If we get here, the basic pipeline worked!
    // Workers pulled contigs, split them, compressed them
    // (Archive writing is TODO, but the core flow works)
}

#[test]
fn test_streaming_queue_stats() {
    let config = StreamingQueueConfig {
        queue_capacity: 1024,
        num_threads: 1,
        verbosity: 0,
        ..Default::default()
    };

    let compressor = StreamingQueueCompressor::with_splitters(
        "/tmp/test_queue_stats.agc",
        config,
        AHashSet::new(),
    )
    .expect("Failed to create compressor");

    let stats = compressor.queue_stats();
    assert_eq!(stats.current_size_bytes, 0);
    assert_eq!(stats.current_items, 0);
    assert_eq!(stats.capacity_bytes, 1024);
    assert!(!stats.is_closed);
}