From 4cd8951b2b2206dd3f7726506d624f00932f5fdd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marcin=20Jab=C5=82o=C5=84ski?= <marcin@jablonski.ml>
Date: Wed, 5 Jun 2019 17:14:15 +0200
Subject: [PATCH] fixed look at plants

---
 env.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/env.py b/env.py
index 11fd76f..7511ed0 100755
--- a/env.py
+++ b/env.py
@@ -92,6 +92,13 @@ def try_move(field, tractor):
         return True
     return False
 
+def test_cords(cord, len1, len2):
+    if cord[0] > -1 and cord[0] < len1 \
+        and cord[1] > -1 and cord[1] < len2:
+        return  True
+    return False
+
+
 def look_at_plats(field, location):
     wsp = [
             (location[0] + 1, location[1]),
@@ -99,9 +106,12 @@ def look_at_plats(field, location):
             (location[0], location[1] - 1),
             (location[0], location[1] + 1)
            ]
+    len1 = len(field)
+    len2 = len(field[0])
+    wsp = [l if test_cords(l, len1, len2) else None for l in wsp]
     plants = []
     for i in wsp:
-        if field[i[0]][i[1]]:
+        if i and field[i[0]][i[1]]:
             plants.append(i)
     return plants