Making integration tests more py.test like