friend_functions.py

This commit is contained in:
njmason2
2025-10-21 08:51:10 -04:00
parent 118c219d09
commit 063ea37fa1
2 changed files with 78 additions and 56 deletions

View File

@@ -9,12 +9,12 @@
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,
# count_people,
# get_email,
# count_favorite_colors,
# people_who_like_color,
# count_people_who_like_color,
get_color_dict,
)
# class TestCountPeople(TestCase):
@@ -49,42 +49,42 @@ from friend_functions import (
# 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 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)
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()