Use unittest.mock if available

Use unittest.mock https://docs.python.org/3/library/unitetst.mock.html
starting from >= Python 3.3