# 11-1
city_funcitons.py
def city_function(city, country): return city + ', ' + country
import unittest from city_functions import city_function class TestCityFunction(unittest.TestCase): def test_city_country(self): format = city_function('Santiago', 'Chile') self.assertEqual(format, 'Santiago, Chile') unittest.main()
# 11-2
city_funcitons.py
def city_function(city, country, population= ''): if population: return city + ', ' + country + ' - ' + 'population' + ' ' + population else: return city + ', ' + country
import unittest from city_functions import city_function class TestCityFunction(unittest.TestCase): def test_city_country(self): format = city_function('Santiago', 'Chile') self.assertEqual(format, 'Santiago, Chile') def test_city_country_population(self): format = city_function('Santiago', 'Chile', '5000000') self.assertEqual(format, 'Santiago, Chile - population 5000000') unittest.main()
# 11-3 import unittest class Employee(): def __init__(self, last_name, first_name, salary): self.last_name = last_name self.first_name = first_name self.salary = salary def inc_salary(self, new_salary = 5000): self.salary += new_salary # print("OK, the salary has been raised ' + str(new_salary)+'.\nNow it's " + str(self.salary) +'.') class Test_Employee(unittest.TestCase): def setUp(self): self.first_name = 'Donald' self.last_name = 'Trump' self.my_Employee = Employee(self.first_name, self.last_name, 10000) def test_give_default_raise(self): self.my_Employee.inc_salary() self.assertEqual(self.my_Employee.salary, 15000) def test_give_custom_raise(self): self.my_Employee.inc_salary(10000) self.assertEqual(self.my_Employee.salary, 20000) unittest.main()