def test_sanitize_date_range_nones() -> None:
yesterday = date.today() - timedelta(days=1)
start_dt, end_dt = sanitize_date_range(None, None)
assert start_dt == yesterday
assert end_dt == date.today()
def test_sanitize_date_range_start_dt_gt_end_dt() -> None:
start_dt = '2020-05-06'
end_dt = '2019-08-01'
start_dt_date, end_dt_date = sanitize_date_range(start_dt, end_dt)
assert start_dt_date < end_dt_date
assert str(start_dt_date) == end_dt
assert str(end_dt_date) == start_dt
def test_handle_request_pre_season() -> None:
start_dt, end_dt = sanitize_date_range('2019-03-01', '2019-03-22')
result = _handle_request(start_dt, end_dt, step=1, verbose=False)
assert result is not None
assert not result.empty
assert len(result.columns) == CURRENT_SC_COLUMNS
assert len(result) > 0
def test_small_request() -> None:
start_dt, end_dt = sanitize_date_range('2019-06-01', None)
result = _small_request(start_dt, end_dt)
assert result is not None
assert not result.empty
assert len(result.columns) == CURRENT_SC_COLUMNS
assert len(result) > 0
def test_handle_request_post_season() -> None:
start_dt, end_dt = sanitize_date_range('2018-11-14', '2019-03-22')
result = _handle_request(start_dt, end_dt, step=1, verbose=False)
assert result is not None
assert not result.empty
assert len(result.columns) == 89
assert len(result) == 689
def test_small_request_auto_chunking() -> None:
start_dt, end_dt = sanitize_date_range('2019-06-01', '2019-06-15')
result = small_request(start_dt, end_dt)
assert result is not None
assert not result.empty
assert len(result.columns) == 89
assert len(result) > _MAX_SC_RESULTS
def test_small_request() -> None:
start_dt, end_dt = sanitize_date_range('2019-06-01', None)
result = small_request(start_dt, end_dt)
assert result is not None
assert not result.empty
assert len(result.columns) == 89
assert len(result) == 4556
def test_handle_request_post_season() -> None:
start_dt, end_dt = sanitize_date_range('2018-11-14', '2019-03-22')
with pytest.warns(UserWarning):
result = _handle_request(start_dt, end_dt, step=1, verbose=False)
assert result is not None
assert not result.empty
assert len(result.columns) == CURRENT_SC_COLUMNS
assert len(result) > 0
def test_handle_request_post_season_same_year() -> None:
start_dt, end_dt = sanitize_date_range('2018-11-14', '2018-11-30')
result = _handle_request(start_dt, end_dt, step=1, verbose=False)
assert result is not None
assert result.empty
def test_sanitize_date_range_bad_end_dt() -> None:
with pytest.raises(ValueError) as ex_info:
sanitize_date_range('2020-05-06', 'INVALID')
assert str(ex_info.value) == 'Incorrect data format, should be YYYY-MM-DD'
def test_sanitize_date_range() -> None:
start_dt, end_dt = sanitize_date_range('2020-05-06', '2020-06-06')
assert start_dt == datetime.strptime('2020-05-06', DATE_FORMAT).date()
assert end_dt == datetime.strptime('2020-06-06', DATE_FORMAT).date()