njmason2/lab_weather
1
0
Fork 0
generated from mwc/lab_weather
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

friend_functions.py

Browse Source
This commit is contained in:
njmason2
2025-10-18 11:22:38 -04:00
parent a5e214b29f
commit 271b54e569
2 changed files with 39 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
friend_functions.py
View File

@@ -36,22 +36,22 @@
# return None # reached the end of the list without a match
def count_favorite_colors(people, name):
"""Returns the number of colors liked by the named person. If there is no such person, returns None.
# def count_favorite_colors(people, name):
# """Returns the number of colors liked by the named person. If there is no such person, returns None.
>>> count_favorite_colors(family, "Tad Winters")
2
>>> count_favorite_colors(family, "Raphael Chambers")
1
"""
for person in people:
if (person["name"]==name): # for a dict within a list lookup
return len(person["favorite_colors"]) # count of list elements in a dict within a list
# >>> count_favorite_colors(family, "Tad Winters")
# 2
# >>> count_favorite_colors(family, "Raphael Chambers")
# 1
# """
# for person in people:
# if (person["name"]==name): # for a dict within a list lookup
# return len(person["favorite_colors"]) # count of list elements in a dict within a list
return None # reached the end of the list without a match
# return None # reached the end of the list without a match
#def people_who_like_color(people, color):
def people_who_like_color(people, color):
"""Returns a list containing only those people who like the given color.
>>> people_who_like_color(family, "yellow")
[
@@ -69,7 +69,16 @@ def count_favorite_colors(people, name):
>>> people_who_like_color(family, "indigo")
[]
"""
# raise NotImplementedError()
fav_name=[] # initialize output list as None
for person in people:
if color in person["favorite_colors"]: # if color in the data file matches input color
fav_name.append(person) # append output list
# print(fav_name)
return fav_name # at the end of file, output list
#def count_people_who_like_color(people, color):
"""Returns the number of people who like a given color.

34
test_friend_functions.py
View File

@@ -11,8 +11,8 @@ from people import friends, family
from friend_functions import (
# count_people,
# get_email,
count_favorite_colors,
# people_who_like_color,
# count_favorite_colors,
people_who_like_color,
# count_people_who_like_color,
# get_color_dict,
)
@@ -30,24 +30,24 @@ from friend_functions import (
# 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)
# 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)
# 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)
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")
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):