11-1
def city_country(city, country): return(city.title() + ", " + country.title())import unittestclass CitiesTestCase(unittest.TestCase): def test_city_country(self): """Does a simple city and country pair work?""" santiago_chile = city_country('santiago', 'chile')
self.assertEqual(santiago_chile, 'Santiago, Chile')
unittest.main()11-3
class Employee(): """A class to represent an employee.""" def __init__(self, f_name, l_name, salary): """Initialize the employee.""" self.first = f_name.title() self.last = l_name.title() self.salary = salary def give_raise(self, amount=5000): """Give the employee a raise.""" self.salary += amountimport unittestclass TestEmployee(unittest.TestCase): """Tests for the module employee.""" def setUp(self): """Make an employee to use in tests.""" self.eric = Employee('eric', 'matthes', 65000) def test_give_default_raise(self): """Test that a default raise works correctly.""" self.eric.give_raise() self.assertEqual(self.eric.salary, 70000) def test_give_custom_raise(self): self.eric.give_raise(10000) self.assertEqual(self.eric.salary, 75000)unittest.main()
