40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
|
from io import BytesIO
|
||
|
|
|
||
|
|
from openpyxl import Workbook
|
||
|
|
|
||
|
|
from bls.parser import parse_components_xlsx, parse_foods_xlsx
|
||
|
|
|
||
|
|
|
||
|
|
def _xlsx(rows):
|
||
|
|
wb = Workbook()
|
||
|
|
ws = wb.active
|
||
|
|
for row in rows:
|
||
|
|
ws.append(row)
|
||
|
|
buf = BytesIO()
|
||
|
|
wb.save(buf)
|
||
|
|
return buf.getvalue()
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_components_dynamic():
|
||
|
|
data = _xlsx([
|
||
|
|
["Code", "Name_DE", "Name_EN", "Einheit"],
|
||
|
|
["ENERCC", "Energie", "Energy", "kcal/100g"],
|
||
|
|
["NA", "Natrium", "Sodium", "mg/100g"],
|
||
|
|
])
|
||
|
|
attrs = parse_components_xlsx(data)
|
||
|
|
keys = {a["attr_key"] for a in attrs}
|
||
|
|
assert "ENERCC" in keys
|
||
|
|
assert "NA" in keys
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_foods_keeps_bls_code():
|
||
|
|
data = _xlsx([
|
||
|
|
["BLS Code", "Name", "Food name", "ENERCC Energie [kcal/100g]", "ENERCC Herkunft", "ENERCC Referenz"],
|
||
|
|
["C131000", "Hafer roh", "Oats raw", 350, "Analyse", "MRI"],
|
||
|
|
])
|
||
|
|
parsed = parse_foods_xlsx(data)
|
||
|
|
assert parsed["foods"][0]["bls_code"] == "C131000"
|
||
|
|
assert parsed["foods"][0]["name_de"] == "Hafer roh"
|
||
|
|
vals = {v["attr_key"]: v["value_num"] for v in parsed["foods"][0]["values"]}
|
||
|
|
assert vals.get("ENERCC") == 350
|