Python Contacts.contacts_by_lives_in Examples

Example #1

File: Test_task7.py Project: NataliiaBidak/py-training-hw

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)