57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
"""Test module `fetch_holidays`. """
|
||
import json
|
||
|
||
import pytest
|
||
|
||
from fetch import (
|
||
CustomJSONEncoder,
|
||
DescriptionParser,
|
||
get_paper,
|
||
get_paper_urls,
|
||
get_rules,
|
||
)
|
||
|
||
from filetools import workspace_path
|
||
|
||
|
||
def test_get_paper_urls():
|
||
assert get_paper_urls(2019) == [
|
||
"http://www.gov.cn/zhengce/content/2018-12/06/content_5346276.htm",
|
||
"http://www.gov.cn/zhengce/content/2019-03/22/content_5375877.htm",
|
||
]
|
||
|
||
|
||
def test_get_rules():
|
||
assert list(
|
||
get_rules(
|
||
get_paper(
|
||
"http://www.gov.cn/zhengce/content/2019-03/22/content_5375877.htm"
|
||
)
|
||
)
|
||
) == [("劳动节", "2019年5月1日至4日放假调休，共4天。4月28日（星期日）、5月5日（星期日）上班。")]
|
||
|
||
|
||
def _normalize(iterable):
|
||
return sorted(
|
||
json.loads(json.dumps(list(iterable), cls=CustomJSONEncoder)),
|
||
key=lambda x: x["date"],
|
||
)
|
||
|
||
|
||
def _description_parsing_cases():
|
||
with open(
|
||
workspace_path("scripts", "description_parsing_cases.json"),
|
||
"r",
|
||
encoding="utf-8",
|
||
) as f:
|
||
return json.load(f)
|
||
|
||
|
||
@pytest.mark.parametrize("case", _description_parsing_cases())
|
||
def test_parse_description(case):
|
||
|
||
year, description, expected = case["year"], case["description"], case["expected"]
|
||
assert _normalize(DescriptionParser(description, year).parse()) == _normalize(
|
||
expected
|
||
), case
|