1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
import json
import os
import unittest
from recipe_scrapers._utils import get_nutrition_keys
class TestNutritionKeyOrder(unittest.TestCase):
def test_nutrition_key_order(self):
nutrition_keys = get_nutrition_keys()
json_directory = "tests/test_data/"
for root, dirs, files in os.walk(json_directory):
for filename in files:
if filename.endswith(".json"):
with open(os.path.join(root, filename), encoding="utf-8") as f:
data = json.load(f)
if "nutrients" in data and data["nutrients"] is not None:
nutrients = data["nutrients"]
actual_keys = list(nutrients.keys())
ordered_keys = [
key for key in nutrition_keys if key in actual_keys
]
other_keys = [
key for key in actual_keys if key not in nutrition_keys
]
expected_keys = ordered_keys + other_keys
try:
self.assertEqual(expected_keys, actual_keys)
except AssertionError as e:
recommend_reorder_helper = (
"\n\nTo automatically reorder the nutrition keys, run the following command:\n"
"python scripts/reorder_json_keys.py\n"
)
raise AssertionError(
f"File: {os.path.join(root, filename)}\n"
f"{str(e)}"
f"{recommend_reorder_helper}"
) from e
|
