Wrapped log testing in a py.test class