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