Add basic testing structure

http://www.reddit.com/r/learnpython/comments/1l1quq/first_real_program_looking_for_refactoring/cbuztw4