[#19][#20] replace unittest tainting by mock functions

Do you need `address_1`, etc., if you have this dict? (You can just loop through the dict's keys.)