Assert Called Once With Python