Import `unittest.mock` if available

Python >= 3.3 has `mock` module builtin as `unittest.mock`. I think it's nice to use it if available