Use py.test rather than pyunit