How to use Mock testing Python