style: use black as formatter

This commit is contained in:
NateScarlet
2021-09-22 19:22:48 +08:00
parent 5ef525db94
commit c79acdc39d
3 changed files with 115 additions and 104 deletions

View File

@@ -1,6 +1,7 @@
"""Tools for files. """
import os
__dirname__ = os.path.abspath(os.path.dirname(__file__))

View File

@@ -3,40 +3,54 @@ import json
import pytest
from fetch_holidays import (CustomJSONEncoder, DescriptionParser, get_paper,
get_paper_urls, get_rules)
from fetch_holidays import (
CustomJSONEncoder,
DescriptionParser,
get_paper,
get_paper_urls,
get_rules,
)
from .filetools import _file_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',
"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日星期日上班。')])
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'])
return sorted(
json.loads(json.dumps(list(iterable), cls=CustomJSONEncoder)),
key=lambda x: x["date"],
)
def _description_parsing_cases():
with open(_file_path('description_parsing_cases.json'), 'r', encoding='utf-8', ) as f:
with open(
_file_path("description_parsing_cases.json"),
"r",
encoding="utf-8",
) as f:
return json.load(f)
@pytest.mark.parametrize('case', _description_parsing_cases())
@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
year, description, expected = case["year"], case["description"], case["expected"]
assert _normalize(DescriptionParser(description, year).parse()) == _normalize(
expected
), case