Replacing test import of mock, using unittest.mock