generated from mwc/lab_weather
91 lines
3.0 KiB
Python
91 lines
3.0 KiB
Python
# 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()
|