在python unittest中创建模拟对象

Question

I am completely new to unittest in Python, so sorry if this is a noob question.

I have a Customer class:

class Customer(object):

    def __init__(self, name, clubMember, taxExempt):
        self.name = name
        self.clubMember = clubMember
        self.taxExempt = taxExempt

    def isClubMember(self):
        if (self.clubMember == True):
            return True
        else:
            return False

    def isTaxExempt(self):
        if (self.taxExempt == True):
            return True
        else:
            return False

And I want to test it with a kind of mock Customer with:

import unittest
import Customer 

class TestCustomer(unittest.TestCase):
    def setUp(self):
        self.customer = Customer("John Doe", True, False)

    def test_customer_member_status(self):
        self.assertTrue(self.customer.isClubMember(), 'incorrect member status')

    def test_customer_tax_exemption(self):
        self.assertFalse(self.customer.isTaxExempt(), 'incorrect tax status')

    def tearDown(self):
        self.customer.dispose()

However, I receive the following error:

self.customer = Customer("John Doe", True, False)
TypeError: 'module' object is not callable

I've tried to search many different websites, but I cannot find the correct format to create the mock Customer. In addition, I'm not sure how to create multiple mock Customers to test another example, such as Customer("Jane Doe", False, True) etc.

Answer 1

I guess your module name is also Customer and when you are doing import Customer , it is importing the module and not the class.

Do this instead

from Customer import Customer 

in test moudle.