Skip to content

Commit

Permalink
added sum trait for Duration
Browse files Browse the repository at this point in the history
  • Loading branch information
jakevossen5 committed Jan 22, 2021
1 parent 3467172 commit 9502e2c
Showing 1 changed file with 33 additions and 0 deletions.
33 changes: 33 additions & 0 deletions src/oldtime.rs
Original file line number Diff line number Diff line change
Expand Up @@ -373,6 +373,18 @@ impl Div<i32> for Duration {
}
}

impl<'a> std::iter::Sum<&'a Duration> for Duration {
fn sum<I: Iterator<Item = &'a Duration>>(iter: I) -> Duration {
iter.fold(Duration::zero(), |acc, x| acc + *x)
}
}

impl std::iter::Sum<Duration> for Duration {
fn sum<I: Iterator<Item = Duration>>(iter: I) -> Duration {
iter.fold(Duration::zero(), |acc, x| acc + x)
}
}

impl fmt::Display for Duration {
/// Format a duration using the [ISO 8601] format
///
Expand Down Expand Up @@ -630,6 +642,27 @@ mod tests {
assert_eq!(Duration::seconds(-4) / -3, Duration::nanoseconds(1_333_333_333));
}

#[test]
fn test_duration_sum() {
let duration_list_1 = [Duration::zero(), Duration::seconds(1)];
let sum_1: Duration = duration_list_1.iter().sum();
assert_eq!(sum_1, Duration::seconds(1));

let duration_list_2 =
[Duration::zero(), Duration::seconds(1), Duration::seconds(6), Duration::seconds(10)];
let sum_2: Duration = duration_list_2.iter().sum();
assert_eq!(sum_2, Duration::seconds(17));

let duration_vec = vec![
Duration::zero(),
Duration::seconds(1),
Duration::seconds(6),
Duration::seconds(10),
];
let sum_3: Duration = duration_vec.into_iter().sum();
assert_eq!(sum_3, Duration::seconds(17));
}

#[test]
fn test_duration_fmt() {
assert_eq!(Duration::zero().to_string(), "PT0S");
Expand Down

0 comments on commit 9502e2c

Please sign in to comment.