class TestContacts(unittest.TestCase): def setUp(self): self.home_book = Contacts() self.work_book = Contacts() def test_create_and_search(self): self.home_book.add('Tom', { 'lives_in': 'USA', 'email': '*****@*****.**' }) self.home_book.add('Bob', { 'lives_in': 'USA', 'email': '*****@*****.**' }) self.work_book.add('Mike', { 'lives_in': 'Marks', 'email': '*****@*****.**' }) results = self.home_book.contacts_by_lives_in('USA') self.assertTrue('Tom' in results) self.assertTrue('Bob' in results) self.assertEqual(len(results), 2)