Unit testing a contact list
The code sample below has Contact class that contains both a Person and an Address class, and finally, a Notebook class that contains multiple contacts.
Can you use pytest and unittest.mock modules to write tests for these classes and fix the bugs in this code
### useful: This is the code you should test
class Address:
def __init__(self, street, city):
self.street = str(street)
self.city = str(city)
def __repr__(self):
return f"Address({self.city!r}, {self.street!r})"
class Person:
def __init__(self, name, email):
self.name = name
self.email= email
def __repr__(self):
return f"Person({self.name!r}, {self.email!r})"
class Contact:
def __init__(self, street, city, name, email, **kwargs):
self.person = Person(name, email)
self.address = Address(street, city)
def __str__(self):
return f"""\
{self.person.name}:
{self.person.email}
address:
{self.address.city}
{self.address.street}
"""
class Notebook:
def __init__(self):
self.contacts = dict()
def add(self, street, city, name, email):
self.contacts[name] = Contact(name, email, city, street)
def remove(name):
self.contacts.remove(name)
def __str__(self):
results = []
for name, contact in self.contacts.items():
results.append(str(contact))
results.append("")
return '\n'.join(results)
# write your tests here
import pytest
import unittest.mock as mocking
@pytest.fixture
def city():
return 'city'
@pytest.fixture
def street():
return 'street'
def test_address(city, street):
address = Address(street=street, city=city)
assert address.street == street
assert address.city == city
@pytest.fixture
def name():
return 'name'
@pytest.fixture
def email():
return 'email'
def test_person(name, email):
person = Person(name=name, email=email)
assert person.name == name
assert person.email == email
def test_contact(name, email, city, street):
contact = Contact(name=name, email=email, city=city, street=street)
assert contact.person.name == name
assert contact.person.email == email
assert contact.address.city == city
assert contact.address.street == street
@pytest.fixture
def empty_notebook():
return Notebook()
def test_empty_notebook(empty_notebook):
assert len(empty_notebook.contacts) == 0
def test_notebook_add(empty_notebook, name, email, city, street):
empty_notebook.add(name=name, email=email, city=city, street=street)
assert len(empty_notebook.contacts) == 1
assert empty_notebook.contacts[name].person.name == name
assert empty_notebook.contacts[name].person.email == email
assert empty_notebook.contacts[name].address.city == city
assert empty_notebook.contacts[name].address.street == street
### useful: run the tests you wrote
import ipytest
# enable pytest's assertions and ipytest's magics
ipytest.config(rewrite_asserts=True, magics=True)
# set the filename
__file__ = 'ex 08 - solutions.ipynb'
# execute the tests via pytest, arguments are passed to pytest
ipytest.run('-qq')