def test_sanitize_input_no_end_dt(self):
yesterday = (datetime.today() - timedelta(days=1)).strftime('%Y-%m-%d')
start_dt, end_dt = sanitize_input(yesterday, None)
assert start_dt == yesterday
assert end_dt == yesterday
def test_sanitize_input_no_end_dt(self) -> None:
yesterday = date.today() - timedelta(days=1)
start_dt, end_dt = sanitize_input(str(yesterday), None)
assert start_dt == yesterday
assert end_dt == yesterday
def test_statcast(
self,
response_get_monkeypatch: Callable,
small_request_raw: str,
small_request: pd.DataFrame
):
start_dt, end_dt = sanitize_input(None, None)
response_get_monkeypatch(
small_request_raw.encode('UTF-8'),
_SC_SMALL_REQUEST.format(
start_dt=start_dt,
end_dt=end_dt
)
)
statcast_result = statcast().reset_index(drop=True)
pd.testing.assert_frame_equal(statcast_result, small_request)
def test_sanitize_input_bad_end_dt(self):
with pytest.raises(ValueError) as ex_info:
sanitize_input('2020-05-06', 'INVALID')
assert str(ex_info.value) == 'Incorrect data format, should be YYYY-MM-DD'
def test_sanitize_input(self):
start_dt, end_dt = sanitize_input('2020-05-06', '2020-06-06')
assert start_dt == '2020-05-06'
assert end_dt == '2020-06-06'
def test_sanitize_input(self) -> None:
start_dt, end_dt = sanitize_input('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()
