def test_period_determiner_29():
# если используется форма "за квартал",
# без указания номера квартала, то имеем ввиду текущий квартал
current_date = datetime.now()
words_to_process = ['квартал']
qvartal_nomer = 1
if current_date.month in [1, 2, 3]:
qvartal_nomer = 1
elif current_date.month in [4, 5, 6]:
qvartal_nomer = 2
elif current_date.month in [7, 8, 9]:
qvartal_nomer = 3
else:
qvartal_nomer = 4
d1 = datetime(year=current_date.year, month=3 * qvartal_nomer - 2, day=1)
d2: Optional[datetime] = None
if qvartal_nomer == 4:
d2 = datetime(year=current_date.year + 1, month=1,
day=1) - timedelta(days=1)
else:
d2 = datetime(year=current_date.year,
month=3 * qvartal_nomer + 1,
day=1) - timedelta(days=1)
result = period_determiner(words_to_process, future_days_allowed=True)
assert result == (d1.strftime('%d.%m.%Y'), d2.strftime('%d.%m.%Y'))
def test_period_determiner_1():
# если используется форма "с месяца года по ныне"
current_date = datetime.now()
words_to_process = ['с', 'марта', '2019', 'года', 'по', 'ныне']
result = period_determiner(words_to_process)
assert result == ('01.03.2019', current_date.strftime('%d.%m.%Y'))
def test_period_determiner_30():
# если даты передана в формате d.m
current_date = datetime.now()
words_to_process = ['8.1']
result = period_determiner(words_to_process)
assert result == ('08.01.{}'.format(current_date.year),
'08.01.{}'.format(current_date.year))
def test_period_determiner_31():
# если используется корректная форма к примеру "с 2019 года",
# то период определится как с начала 2019 года по сегодня
current_date = datetime.now()
words_to_process = ['с', '2019', 'года']
result = period_determiner(words_to_process)
assert result == ('01.01.2019', current_date.strftime('%d.%m.%Y'))
def test_period_determiner_15():
# если используется только месяц
current_date = datetime.now()
words_to_process = ['март']
result = period_determiner(words_to_process)
assert result == ('01.03.{}'.format(current_date.year),
'31.03.{}'.format(current_date.year))
def test_period_determiner_23():
# если используется форма "с 2 по 17 июня"
current_date = datetime.now()
count_of_months: int = 2
words_to_process = ['с', '2', 'по', '17', 'января']
result = period_determiner(words_to_process)
assert result == ('02.01.{}'.format(current_date.year),
'17.01.{}'.format(current_date.year))
def test_period_determiner_14():
# если на обработку попадут слова не связанные с периодами времени
words_to_process = [
'ytgjy',
'до',
'sdfdasq0',
]
result = period_determiner(words_to_process)
assert result == (UNRECOGNIZED_DATE_VALUE, UNRECOGNIZED_DATE_VALUE)
def test_period_determiner_13():
# если передать две даты с союзом "и" между ними, то определится только последняя
words_to_process = [
'31.01.2003',
'и',
'12.03.2020',
]
result = period_determiner(words_to_process)
assert result == ('12.03.2020', '12.03.2020')
def test_period_determiner_12():
# если одна из дат некорректная
words_to_process = [
'34.01.2003',
'до',
'12.03.2020',
]
result = period_determiner(words_to_process)
assert result == (ERROR_VALUE, '12.03.2020')
def test_period_determiner_8():
# если корректные даты переданы в формате dd.mm.yy и dd.mm.yyyy
words_to_process = [
'28.01.12',
'до',
'12.03.2020',
]
result = period_determiner(words_to_process)
assert result == ('28.01.2012', '12.03.2020')
def test_period_determiner_28():
# если используется корректная форма "за неделю"
# без указания количества недель
current_date = datetime.now()
words_to_process = ['неделю']
result = period_determiner(words_to_process)
d1: datetime = current_date - timedelta(7)
assert result == (d1.strftime('%d.%m.%Y'),
current_date.strftime('%d.%m.%Y'))
def test_period_determiner_10():
# если конкретный квартал ипользуется
current_date = datetime.now()
words_to_process = [
'3',
'квартал',
'2020',
]
result = period_determiner(words_to_process)
assert result == ('01.07.2020', '30.09.2020')
def test_period_determiner_26():
# если используется корректная форма к примеру "за последние n дней",
# то период определится как с даты ранее на n дней текущего дня
current_date = datetime.now()
count_of_days: int = 30
words_to_process = ['за', 'последние', str(count_of_days), 'дней']
result = period_determiner(words_to_process)
d1: datetime = current_date - timedelta(count_of_days)
assert result == (d1.strftime('%d.%m.%Y'),
current_date.strftime('%d.%m.%Y'))
def test_period_determiner_6():
# если используется форма "за несколько прошлых дня"
current_date = datetime.now()
delta_day = 4
words_to_process = [str(delta_day), 'прошлых', 'дня']
result = period_determiner(words_to_process)
d = current_date - timedelta(days=delta_day)
assert result == (d.strftime('%d.%m.%Y'),
current_date.strftime('%d.%m.%Y'))
def test_period_determiner_25():
# если используется корректная форма к примеру "за n недель",
# то период определится как с даты ранее на 7 * n дней текущего дня
current_date = datetime.now()
count_of_weeks: int = 3
words_to_process = ['за', str(count_of_weeks), 'недели']
result = period_determiner(words_to_process)
d1: datetime = current_date - timedelta(7 * count_of_weeks)
assert result == (d1.strftime('%d.%m.%Y'),
current_date.strftime('%d.%m.%Y'))
def test_period_determiner_11():
# если начиная с определенного квартала
current_date = datetime.now()
words_to_process = [
'с',
'3',
'квартала',
'2020',
]
result = period_determiner(words_to_process)
assert result == ('01.07.2020', current_date.strftime('%d.%m.%Y'))
def test_period_determiner_18():
# если используется корректная форма к примеру "за 28 марта",
# то определится как 28 марта текущего года
current_date = datetime.now()
words_to_process = [
'28',
'марта',
]
result = period_determiner(words_to_process)
assert result == ('28.03.{}'.format(current_date.year),
'28.03.{}'.format(current_date.year))
def test_period_determiner_4():
# если дата начала больше даты окончания периода
words_to_process = [
'марта',
'2019',
'года',
'по',
'15',
'июня',
'2010',
]
result = period_determiner(words_to_process)
assert result == (ERROR_VALUE, ERROR_VALUE)
def test_period_determiner_3():
# если используется полная форма
words_to_process = [
'марта',
'2019',
'года',
'по',
'15',
'июня',
'2021',
]
result = period_determiner(words_to_process)
assert result == ('01.03.2019', '15.06.2021')
def test_period_determiner_2():
# если используется форма "за месяц год по сегодняшний день"
current_date = datetime.now()
words_to_process = [
'марта',
'2019',
'года',
'по',
'сегодняшний',
'день',
]
result = period_determiner(words_to_process)
assert result == ('01.03.2019', current_date.strftime('%d.%m.%Y'))
def extract(
self,
text_preprocessing_result: TextPreprocessingResult,
user: User,
params: Optional[Dict[str, Union[str, float, int]]] = None
) -> Dict[str, str]:
if text_preprocessing_result\
.words_tokenized_set\
.intersection(
[
'TIME_DATE_TOKEN',
'TIME_DATE_INTERVAL_TOKEN',
'PERIOD_TOKEN'
]):
words_from_intent: List[Optional[
str]] = text_preprocessing_result.human_normalized_text.lower(
).split()
else:
words_from_intent: List[
Optional[str]] = text_preprocessing_result.original_text.lower(
).split()
words_to_process: List[
Optional[str]] = extract_words_describing_period(words_from_intent)
begin_str, end_str = period_determiner(words_to_process,
self.max_days_in_period,
self.future_days_allowed)
is_determined: bool = False
is_error: bool = False
if not (begin_str == '' or begin_str == 'error' or end_str == ''
or end_str == 'error'):
is_determined = True
if begin_str == 'error' or end_str == 'error':
is_error = True
return {
'date_period__is_determined': is_determined,
'date_period__is_error': is_error,
'date_period__begin_date': begin_str,
'date_period__end_date': end_str
}
def test_period_determiner_5():
# если используется форма "за несколько прошлых месяцев"
current_date = datetime.now()
delta_month = 4
words_to_process = [str(delta_month), 'прошлых', 'месяца']
result = period_determiner(words_to_process)
month: int = 0
year: int = 0
if delta_month > current_date.month:
month = 12 + current_date.month - delta_month
year = current_date.year - 1
else:
month = current_date.month - delta_month
year = current_date.year
assert result == ('01.{}.{}'.format(str(month).zfill(2), year),
current_date.strftime('%d.%m.%Y'))
def test_period_determiner_16():
# если используется некорретная форма к примеру за март 28 2020 года
words_to_process = ['март', '28', '2020', 'года']
result = period_determiner(words_to_process)
assert result == (ERROR_VALUE, ERROR_VALUE)
def test_period_determiner_17():
# если используется некорректная форма к примеру "за март 28",
words_to_process = ['март', '28']
result = period_determiner(words_to_process)
assert result == (ERROR_VALUE, ERROR_VALUE)
def test_period_determiner_19():
# если используется корректная форма к примеру "за 28 марта",
# то определится как 28 марта текущего года
words_to_process = ['28', 'марта', '2019']
result = period_determiner(words_to_process)
assert result == ('28.03.2019', '28.03.2019')
def test_period_determiner_7():
# если переданы год и месяц
words_to_process = ['2013', 'июнь']
result = period_determiner(words_to_process)
assert result == ('01.06.2013', '30.06.2013')
def test_period_determiner_24():
# тест контроля максимального количества дней в периоде
count_of_years: int = 2
words_to_process = ['за', str(count_of_years), 'года']
result = period_determiner(words_to_process, max_days_in_period=365)
assert result == (ERROR_VALUE, ERROR_VALUE)
def test_period_determiner_27():
# когда используется только номер года и сам год
words_to_process = ['за', '2020', 'год']
result = period_determiner(words_to_process)
assert result == ('01.01.2020', '31.12.2020')
