def test_make_friends_again(self): temp = FriendShips() temp.makeFriends("Kowalski", "Nowak") temp.makeFriends("Kowalski", "Nowak") self.assertEqual(temp.makeFriends("Kowalski", "Nowak"), 'Already friends!')
def test_make_friends(self): temp = FriendShips() temp.makeFriends("Kowalski", "Nowak") self.assertEqual(temp.data, { "Kowalski": ["Nowak"], "Nowak": ["Kowalski"] })
class FriendShipsStore: def __init__(self): self.friendShips = FriendShips() def makeFriends(self, person1, person2): self.friendShips.makeFriends(person1, person2) def getFriendsList(self, person): self.friendShips.getFriendsList(person) def areFriends(self, person1, person2): self.friendShips.areFriends(person1, person2) def addFriend(self, person, friend): self.friendShips.addFriend(person, friend)
def __init__(self): self.friendShips = FriendShips()
def test_are_friends_error(self): temp = FriendShips() temp.makeFriends("Kowalski", "Nowak") with self.assertRaises(KeyError): temp.areFriends("Kwiatkowski", "Kowalski")
def test_are_friends_false(self): temp = FriendShips() temp.makeFriends("Kowalski", "Nowak") temp.makeFriends("Kowalski", "Kwiatkowski") self.assertFalse(temp.areFriends("Nowak", "Kwiatkowski"))
def test_get_friends_list_error(self): temp = FriendShips() temp.makeFriends("Kowalski", "Nowak") with self.assertRaises(KeyError): temp.getFriendsList("Kwiatkowski")
def test_get_friends_list(self): temp = FriendShips() temp.makeFriends("Kowalski", "Nowak") self.assertEqual(temp.getFriendsList("Kowalski"), ["Nowak"])