The best idea these days is to really use DocTest for the documentation. But I prefer PyDocTest of having test as functions. You can create classes if you want, but you can also have module level things. So there's a way to integrate them.