def test_add_number_to_phonebook(self): phonebook = PhoneBook() number_of_entries = len(phonebook) self.assertEqual(number_of_entries, 0) phonebook.add_contact('Bob', '91 12 54 26') new_number_of_entries = len(phonebook) self.assertEqual(new_number_of_entries, 1) phonebook.add_contact('Alice', '97 625 992') new_new_number_of_entries = len(phonebook) self.assertEqual(new_new_number_of_entries, 2)
def test_add_inconsistent_longer_number(self): phonebook = PhoneBook() phonebook.add_contact('Emergency', '911') exception = PhoneBook.InconsistentEntryError self.assertRaises(exception, phonebook.add_contact, 'Bob', '91 12 54 26')
def test_lookup_name_by_number(self): phonebook = PhoneBook() phonebook.add_contact('Bob', '91 12 54 26') name = phonebook.lookup_name('91125426') bob = 'Bob' self.assertEqual(name, bob)
def test_lookup_number_by_name(self): phonebook = PhoneBook() phonebook.add_contact('Bob', '91 12 54 26') bob_number = phonebook.lookup_number('Bob') bob = '91125426' self.assertEqual(bob, bob_number)