高级编程技术第十一次作业

第十一章 测试代码

11-1 城市和国家

import unittest

def func(city, country):
    return city + ", " + country

class FunctionTest(unittest.TestCase):

    def test_city_country(self):
        fomatted = func("santiago", "chile")
        self.assertEqual(fomatted, "santiago, chile")

unittest.main()

11-2 人口数量

import unittest

def func(city, country, population=1000000):
    return city + ", " + country + ": " + str(population)

class FunctionTest(unittest.TestCase):

    def test_city_country_population(self):
        fomatted = func("santiago", "chile", population=1000)
        self.assertEqual(fomatted, "santiago, chile: 1000")

unittest.main()

11-3 雇员

import unittest

class Employee():

    def __init__(self, name, salary=10000):
        self.name = name
        self.salary = salary

    def give_raise(self, increment=5000):
        self.salary += increment

class TestEmployee(unittest.TestCase):

    def setUp(self):
        self.employee = Employee("Alice")

    def test_give_default_raise(self):
        self.employee.give_raise()
        self.assertEqual(self.employee.salary, 15000)

    def test_give_custom_raise(self):
        self.employee.give_raise(6000)
        self.assertEqual(self.employee.salary, 16000)

unittest.main()