Merge branch 'master' of github.com:jcxl/flock_game

jxcl · Nov 9, 2013 · 0bca4c8 · 0bca4c8
2 parents 9260a08 + fb679d8
commit 0bca4c8
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/tests/bird_test.py b/tests/bird_test.py
@@ -17,9 +17,28 @@ def test_in_angle(self):
         b1 = bird.Bird((0, 0), (0, 1), 0)
         b2 = bird.Bird((0, -5), (0, 0), 1)
         b3 = bird.Bird((-2, -2), (0, 0), 2)
+
+        b_list = [b2, b3]
+        in_angle = b1.birds_in_angle(b_list)
+        self.assertEquals(in_angle[0].bird_id, 2)
+        self.assertEquals(len(in_angle), 1)
 
+    def test_in_angle_2(self):
+        b1 = bird.Bird((0, 0), (-1, -1), 0)
+        b2 = bird.Bird((5, 5), (0, 0), 1)
+        b3 = bird.Bird((-2, -2), (0, 0), 2)
+
         b_list = [b2, b3]
         in_angle = b1.birds_in_angle(b_list)
         self.assertEquals(in_angle[0].bird_id, 2)
         self.assertEquals(len(in_angle), 1)
 
+    def test_in_angle_3(self):
+        b1 = bird.Bird((0, 0), (-1, 0), 0)
+        b2 = bird.Bird((5, 0), (0, 0), 1)
+        b3 = bird.Bird((0, 0), (0, 0), 2)
+
+        b_list = [b2, b3]
+        in_angle = b1.birds_in_angle(b_list)
+        self.assertEquals(in_angle[0].bird_id, 2)
+        self.assertEquals(len(in_angle), 1)