From 6cdcb8a1ce50883cc3e0af5bf89766b910931040 Mon Sep 17 00:00:00 2001 From: a14m Date: Wed, 7 Apr 2021 18:53:18 +0200 Subject: Add the len method to the plane --- CLI/rust/src/game_of_life/simulation/plane.rs | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/CLI/rust/src/game_of_life/simulation/plane.rs b/CLI/rust/src/game_of_life/simulation/plane.rs index 21c34cf..11d7ee5 100644 --- a/CLI/rust/src/game_of_life/simulation/plane.rs +++ b/CLI/rust/src/game_of_life/simulation/plane.rs @@ -8,6 +8,24 @@ use std::ops::{Index, IndexMut}; #[derive(Debug, Clone, PartialEq)] pub(crate) struct Plane(pub Vec); +impl Plane { + pub fn len(&self) -> isize { + self.0.len() as isize + } +} + +#[cfg(test)] +mod len { + use super::*; + + #[test] + fn get_len() { + assert_eq!(Plane(vec![false, false, true, false, false]).len(), 5); + assert_eq!(Plane(vec![false, false, true, false]).len(), 4); + assert_eq!(Plane(vec![false, false, true]).len(), 3); + } +} + impl Index for Plane { type Output = T; fn index(&self, idx: isize) -> &T { -- cgit v1.2.3