python unit test mock database