Unit tests

test.py

import unittest

from faker import Faker

from customer import Customer


class TestCustomer(unittest.TestCase):

    def setUp(self):
        fake = Faker()
        self.customer = Customer(
                name = fake.company(),
                address = fake.street_address(),
                state = fake.state(),
                zipcode = fake.zipcode(),
                contact_name = fake.name_female(),
                phone_number = fake.phone_number(),
                email = fake.company_email(),
                notes = fake.text(max_nb_chars=20)
        )

    def test_full_address(self):
        expected = f'{self.customer.address} {self.customer.state} {self.customer.zipcode}'
        self.assertEqual(expected, self.customer.full_address)

    def test_contact(self):
        expected = f'{self.customer.contact_name} <self.customer.email>'