Use pytest for testing

py.test gives a little bit nicer output and more options for controlling running the tests.