def test_company_dict(): """Tests company_dict.""" obj = mock.Mock( id=123, name='Name', alias='Trading', spec_set=('id', 'name', 'alias'), ) res = dict_utils.company_dict(obj) assert res == { 'id': str(obj.id), 'name': obj.name, 'trading_name': obj.alias, }
def test_company_dict(obj, expected_dict): """Tests for the company_dict function.""" company = dict_utils.company_dict(obj) assert company == expected_dict
def _companies_list(companies): return [dict_utils.company_dict(company) for company in companies.all()]
def _get_company_list(col): return [dict_utils.company_dict(c) for c in col.all()]