def test_drange():
assert (drange('20200501', '20200531') == [
'202005{:02d}'.format(d) for d in range(1, 31 + 1)
])
assert (drange(
'20200401',
'20200531') == (['202004{:02d}'.format(d) for d in range(1, 30 + 1)] +
['202005{:02d}'.format(d) for d in range(1, 31 + 1)]))
def test_drange():
assert drange("20200501", "20200531") == [
"202005{:02d}".format(d) for d in range(1, 31 + 1)
]
assert drange(
"20200401",
"20200531") == (["202004{:02d}".format(d) for d in range(1, 30 + 1)] +
["202005{:02d}".format(d) for d in range(1, 31 + 1)])
def _prepare_nav_urls(args: argparse.Namespace) -> List[str]:
return [
f"https://news.naver.com/main/list.nhn?mode=LSD&mid=shm"
f"&sid1={category}&date={date}&page={page}"
for category in args.category for date in utils.drange(
args.start_date, args.end_date, args.skip_days)
for page in range(1, args.max_page + 1)
]
def test_drange_creates_correct_date_range():
# Test the range between 2020.5.1 and 2020.5.31.
start, end = '20200501', '20200531'
expected = ['202005{:02d}'.format(day) for day in range(1, 31 + 1)]
assert drange(start, end) == expected