Using unittest.mock instead of mock

Using unittest.mock instead of mock #754