diff --git a/src/playfield.rs b/src/playfield.rs index 5027262..08692cb 100644 --- a/src/playfield.rs +++ b/src/playfield.rs @@ -53,7 +53,7 @@ impl Playfield { } } pub fn get_column(&self, index: usize) -> Vec { - Vec::new() + self.fields.iter().map(|row| row[index].clone()).collect() } } @@ -108,9 +108,9 @@ mod tests { let expected_2 = vec![Field::new(2); 3]; let expected_3 = vec![Field::new(3); 3]; - let column_1 = playfield.get_column(1); - let column_2 = playfield.get_column(2); - let column_3 = playfield.get_column(3); + let column_1 = playfield.get_column(0); + let column_2 = playfield.get_column(1); + let column_3 = playfield.get_column(2); assert_eq!(column_1, expected_1); assert_eq!(column_2, expected_2);