40 lines
891 B
Rust
40 lines
891 B
Rust
use std::ops::Range;
|
|
|
|
pub fn gen_chunks(n: usize, chunks: usize) -> Vec<Range<usize>> {
|
|
let mut borders = vec![];
|
|
let chunk_size = n / chunks;
|
|
let mut start = 0;
|
|
for _ in 0..chunks {
|
|
let end = start + chunk_size;
|
|
borders.push(start..end);
|
|
start = end;
|
|
}
|
|
// Change the last chunk to include the remainder
|
|
let size = borders.len();
|
|
borders[size - 1].end = n;
|
|
borders
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_gen_chunks() {
|
|
let borders = gen_chunks(10, 3);
|
|
assert_eq!(borders, vec![(0..3), (3..6), (6..10)]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_gen_chunks2() {
|
|
let borders = gen_chunks(10, 2);
|
|
assert_eq!(borders, vec![(0..5), (5..10)]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_gen_chunks3() {
|
|
let borders = gen_chunks(10, 1);
|
|
assert_eq!(borders, vec![(0..10)]);
|
|
}
|
|
}
|