11-1
city_functions.py:
def city_country(city,country):
return city+', '+country
test_cities.py:
import unittest
from city_functions import city_country
class city_Test_class(unittest.TestCase):
def test_city_country(self):
String = city_country('Beijing', 'China')
self.assertEqual(String,'Beijing, China')
unittest.main()
11-3
employee.py:
class Employee():
def __init__(self, name, surname, money):
self.property={'name':name,'surname':surname,'money':money}
def give_raise(self,num = 5000):
self.property['money'] += num
test_employee.py:
from employee import Employee
import unittest
class Money_test_class(unittest.TestCase):
def setUp(self):
self.employee = Employee('jack','q',3000)
def test_give_default_raise(self):
self.employee.give_raise()
self.assertEqual(8000, self.employee.property['money'])
def test_give_custom_raise(self):
self.employee.give_raise(2000)
self.assertEqual(5000, self.employee.property['money'])
unittest.main()