class PhonebookEditTestCase(unittest.TestCase):
def setUp(self):
self.phonebook = Phonebook()
self.contact = self.phonebook.add_contact("Redlion", "0715846586")
def test_edit_contact_number(self):
resp = self.phonebook.edit_contact("Redlion", "0754681222")
self.assertNotEqual(resp, "0715846586")
def test_edit_contact_name(self):
resp = self.phonebook.edit_contact("Redlion", "Isaac")
self.assertEqual(resp, None)
