python学习第十一章测试代码部分课后练习自己尝试的代码

11-1：

def city_functions(city,country):
    return city+","+country
import unittest
class Test_City(unittest.TestCase):
    def test_city_country(self):
        string = city_functions("Santiago","Chile")
        self.assertEqual(string,"Santiago,Chile")

unittest.main()

11-2：

def city_functions(city,country,population=5000000):
    return city+","+country+" - population "+str(population)
import unittest
class Test_City(unittest.TestCase):
    def test_city_country_population(self):
        string = city_functions("Santiago","Chile")
        self.assertEqual(string,"Santiago,Chile - population 5000000")
    

unittest.main()

11-3：

class Employee():
    def __init__(self,last_name,first_name,salary):
        self.first_name = first_name
        self.last_name = last_name
        self.salary = salary
    def give_raise(self,add = 5000):
        self.salary += add


import unittest
class Test_Employee(unittest.TestCase):
    def setUp(self):
        self.my_employee = Employee("san","zhang",100)
        self.first_add_salary = 5100
        self.second_add_salary = 1100
    def test_give_default_raise(self):
        self.my_employee.give_raise()
        self.assertEqual(self.my_employee.salary,self.first_add_salary)
    def test_give_custom_raise(self):
        self.my_employee.give_raise(1000)
        self.assertEqual(self.my_employee.salary,self.second_add_salary)


unittest.main()