Python学习笔记10（测试代码）

python模块unittest中的工具可以测试代码

断言方法

assertEqual(a, b) --------- 核实a == b
assertNotEqual(a, b) --------- 核实a != b
assertTrue(x) --------------- 核实x为True

assertFalse(x) -------------- 核实x为False
assertIn(item, list) ---------- 核实item在list中
assertNotIn(item, list) --------- 核实item不在list中

---

employee.py

class Employee():
    """员工信息"""

    def __init__(self, first_name, last_name, yearly_salary):
        """初始化属性"""
        self.first_name = first_name
        self.last_name = last_name
        self.yearly_salary = yearly_salary

    def give_raise(self, raise_num=5000):
        """默认给年薪增长5000"""
        self.yearly_salary += raise_num

test_employee.py

import unittest
from employee import Employee

class RaiseSalaryTest(unittest.TestCase):
    """测试年薪增加的类"""
    def setUp(self):
        self.employee = Employee('陈', '佳禾', 200000)

    def test_give_default_raise(self):
        """测试默认增加5000"""
        self.employee.give_raise()
        self.assertEqual(self.employee.yearly_salary, 205000)

    def test_give_custom_raise(self):
        """测试自定义增加的年薪"""
        self.employee.give_raise(4444)
        self.assertEqual(self.employee.yearly_salary, 204444)

if __name__ == '__main__':
    unittest.main()