Leave it to py.test discovery

Fixes #333.