drop.rs

//! Creates a slice of `array` with `n` elements dropped from the beginning.
//!
//! Example
//! ```
//! use lodash_rust::drop;
//!
//! let res = drop::new([1, 2, 3].to_vec(), Some(5));
//! println!("{res:?}") // []
//! ```

use std::convert::TryInto;

pub fn new<T: PartialOrd + Eq + Clone>(array: Vec<T>, number: Option<i32>) -> Vec<T> {
    // defaults to 1
    let num = number.unwrap_or(1);
    let arr_len = array.len().try_into().unwrap();
    if num > arr_len {
        Vec::new()
    } else {
        array[num as usize..].to_vec()
    }
}

#[test]
fn test_new() {
    assert_eq!(new([1, 2, 3].to_vec(), None), [2, 3]);
    assert_eq!(new([1, 2, 3].to_vec(), Some(2)), [3]);
    assert_eq!(new([1, 2, 3].to_vec(), Some(5)), []);
    assert_eq!(new([1, 2, 3].to_vec(), Some(0)), [1, 2, 3]);
}