Initial commit

test_friend_functions.py

@@ -0,0 +1,90 @@
+# test_friend_functions.py
+# ------------
+# By MWC Contributors
+# 
+# Defines tests for `friend_functions`. 
+# Run this program with `python test_friend_functions.py`.
+# You don't need to edit thie file.
+
+from unittest import TestCase, main
+from people import friends, family
+from friend_functions import (
+    count_people, 
+    get_email, 
+    count_favorite_colors, 
+    people_who_like_color, 
+    count_people_who_like_color,
+    get_color_dict,
+)
+
+class TestCountPeople(TestCase):
+    def test_returns_correct_value(self):
+        self.assertEqual(count_people(family), 5)
+        self.assertEqual(count_people(friends), 10)
+
+class TestGetEmail(TestCase):
+    def test_returns_email_when_person_found(self):
+        self.assertEqual(get_email(family, "Perry Calderon"), "mus.donec@outlook.org")
+        self.assertEqual(get_email(friends, "Joy Tate"), "risus.a.ultricies@hotmail.org")
+
+    def test_returns_none_when_person_missing(self):
+        self.assertEqual(get_email(family, "Ken Kesey"), None)
+
+class TestCountFavoriteColors(TestCase):
+    def test_returns_len_favorite_colors_when_person_found(self):
+        self.assertEqual(count_favorite_colors(family, "Tad Winters"), 2)
+        self.assertEqual(count_favorite_colors(friends,"Connor Rodriguez"), 4)
+        self.assertEqual(count_favorite_colors(friends, "Yuli Reynolds"), 0)
+
+    def test_returns_none_when_person_missing(self):
+        self.assertEqual(count_favorite_colors(family, "Ken Kesey"), None)
+
+class TestPeopleWhoLikeColor(TestCase):
+    def test_returns_correct_length(self):
+        self.assertEqual(len(people_who_like_color(family, "blue")), 2)
+        self.assertEqual(len(people_who_like_color(family, "purple")), 0)
+
+    def test_returns_full_dicts(self):
+        result = people_who_like_color(family, "orange")
+        self.assertEqual(len(result), 1)
+        self.assertEqual(result[0]["email"], "auctor.odio@icloud.ca")
+
+class TestCountPeopleWhoLikeColor(TestCase):
+    def test_returns_correct_count(self):
+        self.assertEqual(count_people_who_like_color(friends, "grey"), 2)
+        self.assertEqual(count_people_who_like_color(friends, "beige"), 1)
+        self.assertEqual(count_people_who_like_color(friends, "cornflower"), 0)
+
+class TestGetColorDict(TestCase):
+    def test_returns_correct_dict(self):
+        familyExpected = {
+            "aqua": 2,
+            "red": 2,
+            "blue": 2,
+            "black": 2,
+            "white": 1,
+            "grey": 1,
+            "yellow": 2,
+            "orange": 1,
+        }
+        friendsExpected = {
+            "lime": 1, 
+            "grey": 2, 
+            "sea foam": 1, 
+            "aqua": 2, 
+            "teal": 1, 
+            "white": 1, 
+            "red": 2, 
+            "black": 2, 
+            "blue": 2, 
+            "tan": 1, 
+            "sand": 1, 
+            "green": 1, 
+            "beige": 1, 
+            "turquoise": 1, 
+            "yellow": 2,
+        }
+        self.assertEqual(get_color_dict(family), familyExpected)
+        self.assertEqual(get_color_dict(friends), friendsExpected)
+
+main()