11-1 城市和国家
import unittest
from city_funtions import get_city_country
class NameTestCase(unittest.TestCase):
def test_city_country(self):
formatted_name = get_city_country('santiago','chile')
self.assertEqual(formatted_name,'Santiago , Chile')
if __name__ == '__main__':
unittest.main()
输出:
Ran 1 test in 0.004s
OK
11-2 人口数量
import unittest
from city_funtions import get_city_country
class NameTestCase(unittest.TestCase):
def test_city_country(self):
formatted_name = get_city_country('santiago','chile')
self.assertEqual(formatted_name,'Santiago , Chile')
def test_city_country_population(self):
formatted_name = get_city_country('Santiago', 'chile', 5000000)
self.assertEqual(formatted_name, "Santiago , Chile - population 5000000")
if __name__ == '__main__':
unittest.main()
输出:
Ran 1 test in 0.001s
OK
11-3 雇员:
#Employee.py
class Employee():
def __init__(self, fir, las, sal = 0):
self.first = fir
self.last = las
self.salary = sal
def give_raise(self, num = 0):
if num:
self.salary += num
else:
self.salary += 5000
#test_Employee.py
import unittest
from Employee import Employee
class TestEmpployee(unittest.TestCase):
def setUp(self):
self.new_employee = Employee("Alex", "Leo")
self.salary = 5000
def test_give_default_raise(self):
self.new_employee.give_raise()
self.assertEqual(self.salary, self.new_employee.salary)
def test_give_custom_raise(self):
add_salary = 4000
self.new_employee.give_raise(add_salary)
self.assertEqual(add_salary + self.salary,self.new_employee.salary)
if __name__ == '__main__':
unittest.main()
输出:
Ran 1 test in 0.001s
OK