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"])