创建单位

Question

嘿，我对python还很陌生，我正在尝试为一些代码创建一个单元测试，我遇到了很多麻烦，说实话，我想知道代码是否是可测试的

""" 
Core employee class
"""


class Employee(object):
empCount = 0

def __init__(self, name, salary, debt, takehome):
    self.name = name
    self.salary = salary
    self.debt = debt
    self.takehome = takehome

def display_employee(self):
    print "Name : ", self.name, ",Salary: ", self.salary, "Debt: ", \
        self.debt, "Take Home: ", self.takehome


emp1 = Employee("Scott", 2000, 200, 2000-200)

emp2 = Employee("Mary", 5000, 300, 5000-300)

emp3 = Employee("Sam", 4000, 700, 4000-700)

emp4 = Employee("Sarah", 7000, 2000, 7000-200)

emp5 = Employee("Charlie", 10000, 5000, 10000-5000)

emp6 = Employee("Tony", 16000, 20000, 16000-20000)


employees = [emp1, emp2, emp3, emp4, emp5, emp6]


for employee in employees:
    employee.display_employee()


Employee.empCount = len(employees)

print "Total Employees is %d" % Employee.empCount}

如果有人能帮我一把，我将不胜感激，如果我需要修改代码，使它可以更容易地运行单元测试，还有什么提示吗

Answer 1

我认为你的课是可以测试的。但是测试是微不足道的，因为你只有一个ˋ__初始化方法和显示方法

单元测试的起点是使用unittest库

为每个“public”方法编写一个测试函数：

class TestEmployee(unittest.TestCase):
    def test_init(self):
        ...

    def test_display_employee(self):
        ...

网上有很多教程，看看这个：Testing Your Code

另见：Python testing