Impl: get_column for playfield

This commit is contained in:
Jonas Zeunert
2024-02-20 00:03:41 +01:00
parent 7eb1d30162
commit 780ae5b956

View File

@@ -53,7 +53,7 @@ impl Playfield {
} }
} }
pub fn get_column(&self, index: usize) -> Vec<Field> { pub fn get_column(&self, index: usize) -> Vec<Field> {
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_2 = vec![Field::new(2); 3];
let expected_3 = vec![Field::new(3); 3]; let expected_3 = vec![Field::new(3); 3];
let column_1 = playfield.get_column(1); let column_1 = playfield.get_column(0);
let column_2 = playfield.get_column(2); let column_2 = playfield.get_column(1);
let column_3 = playfield.get_column(3); let column_3 = playfield.get_column(2);
assert_eq!(column_1, expected_1); assert_eq!(column_1, expected_1);
assert_eq!(column_2, expected_2); assert_eq!(column_2, expected_2);