Adds test file - run with py.test
Adds test file - run with py.test
Fixes issue #4