def test_client_nutrient_report_raw(self, apimock):
cli = UsdaClient("API_KAY")
with apimock:
data = list(cli.get_nutrient_report_raw(nutrients=[42, 1337]))
assert data == NUTRIENT_REPORT_DATA['report']['foods']
def test_client_food_report_v2_error(self, apimock):
cli = UsdaClient("API_KAY")
with apimock:
with pytest.raises(DataGovApiError):
cli.get_food_report_v2(666)
def test_client_food_report_v2(self, apimock):
cli = UsdaClient("API_KAY")
with apimock:
fr = cli.get_food_report_v2(123456)
assert fr[0].food.name == "Pizza"
def test_client_food_report_v2_raw(self, apimock):
cli = UsdaClient("API_KAY")
with apimock:
data = cli.get_food_report_v2_raw(ndbno=123456)
assert data == FOOD_REPORT_V2_DATA
def test_client_list_derivation_codes(self, apimock):
cli = UsdaClient("API_KAY")
with apimock:
codes = list(cli.list_derivation_codes(5))
assert codes[0].name == "Analytical data"
assert codes[1].name == "Analytical data; derived by linear regression"
def test_client_list_derivation_codes_raw(self, apimock):
cli = UsdaClient("API_KAY")
with apimock:
codes = list(cli.list_derivation_codes_raw(max=5))
assert codes == DERIVATION_CODES_LIST_DATA['list']['item']
