add py.test support (fixes #1)