class TestPhonebook(unittest.TestCase): def setUp(self): self.new = Phonebook() def test_create(self): self.new.create() self.assertEqual(self.new.create_flag, True) def test_add(self): self.new.add("James Ray", "123456789") self.new.add("Laura Rose", "23456") self.new.add("Ambulance", "911") self.assertEqual(len(set(self.new.database.hashkeys)), 4) self.assertEqual(self.new.database["Laura Rose"], "23456") def test_remove(self): self.new.create() self.new.add("James Ray", "123456789") self.new.add("Laura Rose", "23456") self.new.add("Ambulance", "911") self.new.remove("James Ray") self.assertRaises(KeyError) def test_change(self): self.new.create() self.new.add("James Ray", "123456789") self.new.add("Laura Rose", "23456") self.new.add("Ambulance", "911") self.new.change("Laura Rose", "9999999") self.assertEqual(self.new.database["Laura Rose"], "9999999") def test_lookup(self): self.new.create() self.new.add("Laura Rose", "23456") self.assertEqual(self.new.lookup("Laura Rose"), "23456") def test_reverse_lookup(self): self.new.create() self.new.add("Laura Rose", "23456") self.assertEqual(self.new.reverse_lookup("23456"), "Laura Rose") def test_save(self): self.new.create() self.new.add("James Ray", "123456789") self.new.add("Laura Rose", "23456") self.new.add("Ambulance", "911") self.new.save() self.new2 = Phonebook() self.assertEqual(self.new.reverse_lookup("23456"), "Laura Rose")