第六周（11章）

11-1：城市和国家

city_funtions.py

def country_and_city(city, country):
    return city.title() + ', '+ country.title();

test_cites.py

from city_funtions import country_and_city
import unittest
class CityTestCase(unittest.TestCase):
    def test_city(self):
        result = country_and_city('santiago', 'chile')
        self.assertEqual(result, 'Santiago, Chile')
unittest.main()

程序运行结果为

.
----------------------------------------------------------------------
Ran 1 test in 0.001s

OK


------------------
(program exited with code: 0)

11-2人口数量

city_funtions.py

def country_and_city(city, country):
    return city.title() + ', '+ country.title() + ' - ' + str(population);

test_cites.py

from city_funtions import country_and_city
import unittest
class CityTestCase(unittest.TestCase):
    def test_city_country(self):
        result = country_and_city('santiago', 'chile')
        self.assertEqual(result, 'Santiago, Chile')
unittest.main()

运行结果为

city_funtions.py改为

def country_and_city(city, country , population=''):
    if population:
        return city.title() + ', '+ country.title() + ' - ' + str(population);
    else :
        return city.title() + ', '+ country.title()

运行结果为

test_cites.py改为

from city_funtions import country_and_city
import unittest

class CityTestCase(unittest.TestCase):
    def test_city(self):
        result = country_and_city('santiago', 'chile')
        self.assertEqual(result, 'Santiago, Chile')
    def test_city_country_population(self):
        result = country_and_city('santiago', 'chile', 5000)
        self.assertEqual(result, 'Santiago, Chile - 5000')
unittest.main()

运行结果为

11-3 雇员

employee.py

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

test_employee.py

from employee import Employee
import unittest

class TestEmployee(unittest.TestCase):
    def setUp(self):
        self.employee = Employee('Hua' , 'Liao', 10000)

    def test_give_deafault(self):
        self.employee.give_raise()
        self.assertEqual(self.employee.money, 15000)
        
    def test_give_custom_raise(self):
        self.employee.give_raise(10000)
        self.assertEqual(self.employee.money, 20000)
        
unittest.main()

运行结果