From 780ae5b95681eaae8e5fb9e78b8583ecb66f9337 Mon Sep 17 00:00:00 2001
From: Jonas Zeunert <jonas@zeunert.org>
Date: Tue, 20 Feb 2024 00:03:41 +0100
Subject: [PATCH] Impl: get_column for playfield

---
 src/playfield.rs | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

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<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_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);