use unittest.mock instead of mock

Could you please consider to utilize built-in unittest.mock?