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()