2217
Comment:
|
5159
convert unit-testing example to unittest
|
Deletions are marked like this. | Additions are marked like this. |
Line 1: | Line 1: |
These are 10 small Python programs. Please keep programs ordered by size. | These Python examples follow the convention that each program gets one line longer than the one before it. Please try to maintain this convention. I also try to introduce at least one new feature in each program. Also, please see note at the bottom. |
Line 3: | Line 3: |
------ | These examples assume version 2.4 or above of Python. {{{ ------ 1 Output |
Line 6: | Line 9: |
------ for name in ('peter', 'paul', 'mary'): |
------ 2 Looping for name in ['peter', 'paul', 'mary']: |
Line 10: | Line 13: |
------ | ------ 3 Input, comments |
Line 15: | Line 18: |
------ | ------ 4 Fibonacci, tuple assignment |
Line 21: | Line 24: |
------ 5 Functions def greet(name): print 'hello', name greet('Jack') greet('Jill') greet('Bob') |
|
Line 22: | Line 31: |
------ # def defines a method in Python def tax(item_charge, g = 0.05): return item_charge * g print '%.2f' % tax(11.35) print '%.2f' % tax(40.00, 0.08) ------ |
------ 6 Import, regular expresssions |
Line 38: | Line 39: |
------ | ------ 7 Dictionaries, generator comprehensions |
Line 43: | Line 44: |
grocery_bill = sum([prices[fruit] * my_purchase[fruit] for fruit in my_purchase]) |
grocery_bill = sum(prices[fruit] * my_purchase[fruit] for fruit in my_purchase) |
Line 48: | Line 49: |
------ class ShoppingCart: def __init__(self): self.items = [] def buy(self, item): self.items.append(item) def boughtItems(self): return self.items my_cart = ShoppingCart() my_cart.buy('apple') my_cart.buy('banana') print my_cart.boughtItems() |
------ 8 Command line arguments, exception handling #!/usr/local/bin/python # This program adds up integers in the command line import sys try: total = sum(int(arg) for arg in sys.argv[1:]) print 'sum =', total except ValueError: print 'Please supply integer arguments' |
Line 59: | Line 60: |
------ | ------ 9 Opening files |
Line 70: | Line 71: |
------ | ------ 10 Time, conditionals |
Line 81: | Line 82: |
------ 11 Triple-quoted strings, while loop REFRAIN = ''' %d bottles of beer on the wall, %d bottles of beer, take one down, pass it around, %d bottles of beer on the wall! ''' bottles_of_beer = 99 while bottles_of_beer > 1: print REFRAIN % (bottles_of_beer, bottles_of_beer, bottles_of_beer - 1) bottles_of_beer -= 1 ------ 12 List slicing def sieve_of_eratosthenes(candidates): i = 0 while True: divisor = candidates[i] if divisor * divisor > candidates[-1]: return candidates else: i += 1 candidates = candidates[:i] + \ [num for num in candidates[i:] if num % divisor != 0] print sieve_of_eratosthenes(range(2,100)) ------ 13 Unit testing # Let's write reusable code, and unit test it. def add_money(amounts): # do arithmetic in pennies so as not to accumulate float errors pennies = sum([round(int(amount * 100)) for amount in amounts]) return float(pennies / 100.0) if __name__ == '__main__': import unittest class TestAddMoney(unittest.TestCase): def test_float_errors(self): self.failUnlessEqual(add_money([0.13, 0.02]), 0.15) self.failUnlessEqual(add_money([100.01, 99.99]), 200) self.failUnlessEqual(add_money([0, -13.00, 13.00]), 0) unittest.main() ------ 14 Classes class BankAccount: def __init__(self, initial_balance = 0): self.balance = initial_balance def deposit(self, amount): self.balance += amount def withdraw(self, amount): self.balance -= amount def overdrawn(self): return self.balance < 0 my_account = BankAccount() my_account.deposit(15) my_account.withdraw(5) print my_account.balance print my_account.overdrawn() }}} Hi, I started this page in May 2007, and I provided the first 10+ or so examples (which may have changed since then). -- SteveHowell All code on this page is open source, of course, with the standard Python license. Minor cleanups are welcome, but if you want to do major restructuring of this page, please contact me through the Python mailing list, or if you are impatient for a response, please just make your own copy of this page. Thanks, and I hope this code is useful for you! |
These Python examples follow the convention that each program gets one line longer than the one before it. Please try to maintain this convention. I also try to introduce at least one new feature in each program. Also, please see note at the bottom.
These examples assume version 2.4 or above of Python.
------ 1 Output print 'hello world' ------ 2 Looping for name in ['peter', 'paul', 'mary']: print name ------ 3 Input, comments # This is a Python comment. \n is a newline name = raw_input('What is your name?\n') print 'Hi', name ------ 4 Fibonacci, tuple assignment parent_rabbits, baby_rabbits = (1, 1) while baby_rabbits < 100: print 'This generation has %d rabbits' % baby_rabbits parent_rabbits, baby_rabbits = (baby_rabbits, parent_rabbits + baby_rabbits) ------ 5 Functions def greet(name): print 'hello', name greet('Jack') greet('Jill') greet('Bob') ------ 6 Import, regular expresssions import re for test_string in [ '555-1212', 'ILL-EGAL']: if re.match('\d\d\d-\d\d\d\d$', test_string): print test_string, 'is a valid US local phone number' else: print test_string, 'rejected' ------ 7 Dictionaries, generator comprehensions prices = {'apple': 0.40, 'banana': 0.50} my_purchase = { 'apple': 1, 'banana': 6} grocery_bill = sum(prices[fruit] * my_purchase[fruit] for fruit in my_purchase) print 'I owe the grocer $%.2f' % grocery_bill ------ 8 Command line arguments, exception handling #!/usr/local/bin/python # This program adds up integers in the command line import sys try: total = sum(int(arg) for arg in sys.argv[1:]) print 'sum =', total except ValueError: print 'Please supply integer arguments' ------ 9 Opening files # indent your Python code to put into an email import glob python_files = glob.glob('*.py') python_files.sort() for fn in python_files: print ' ------' for line in open(fn): print ' ' + line.rstrip() print ------ 10 Time, conditionals import time now = time.localtime() hour = now.tm_hour if hour < 8: print 'sleeping' elif hour < 9: print 'commuting' elif hour < 17: print 'working' elif hour < 18: print 'commuting' elif hour < 20: print 'eating' elif hour < 22: print 'resting' else: print 'sleeping' ------ 11 Triple-quoted strings, while loop REFRAIN = ''' %d bottles of beer on the wall, %d bottles of beer, take one down, pass it around, %d bottles of beer on the wall! ''' bottles_of_beer = 99 while bottles_of_beer > 1: print REFRAIN % (bottles_of_beer, bottles_of_beer, bottles_of_beer - 1) bottles_of_beer -= 1 ------ 12 List slicing def sieve_of_eratosthenes(candidates): i = 0 while True: divisor = candidates[i] if divisor * divisor > candidates[-1]: return candidates else: i += 1 candidates = candidates[:i] + \ [num for num in candidates[i:] if num % divisor != 0] print sieve_of_eratosthenes(range(2,100)) ------ 13 Unit testing # Let's write reusable code, and unit test it. def add_money(amounts): # do arithmetic in pennies so as not to accumulate float errors pennies = sum([round(int(amount * 100)) for amount in amounts]) return float(pennies / 100.0) if __name__ == '__main__': import unittest class TestAddMoney(unittest.TestCase): def test_float_errors(self): self.failUnlessEqual(add_money([0.13, 0.02]), 0.15) self.failUnlessEqual(add_money([100.01, 99.99]), 200) self.failUnlessEqual(add_money([0, -13.00, 13.00]), 0) unittest.main() ------ 14 Classes class BankAccount: def __init__(self, initial_balance = 0): self.balance = initial_balance def deposit(self, amount): self.balance += amount def withdraw(self, amount): self.balance -= amount def overdrawn(self): return self.balance < 0 my_account = BankAccount() my_account.deposit(15) my_account.withdraw(5) print my_account.balance print my_account.overdrawn()
Hi, I started this page in May 2007, and I provided the first 10+ or so examples (which may have changed since then). -- SteveHowell
All code on this page is open source, of course, with the standard Python license.
Minor cleanups are welcome, but if you want to do major restructuring of this page, please contact me through the Python mailing list, or if you are impatient for a response, please just make your own copy of this page. Thanks, and I hope this code is useful for you!