Refactor tests #4583

remove all the set_context for company.