Impl: get_column for playfield
This commit is contained in:
@@ -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);
|
||||||
|
|||||||
Reference in New Issue
Block a user