Create tests

[philshem]

Would be helpful to have some testing code, especially before releasing to PyPi (see issue).

It's slightly complicated because play_puzzle.py has lots of user input, but some good solutions here.