klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 13:18:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add, test date filter regex and date parsing to get natural date range

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2022-07-14 16:47:32 +04:00
parent b54588717f
commit 4a201d52af
2 changed files with 98 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
tests/test_date_filter.py Normal file
View File

@@ -0,0 +1,44 @@
# Standard Packages
import re
from datetime import timedelta, datetime
# Application Packages
from src.search_filter import date_filter
def test_parse():
test_now = datetime(1984, 4, 1, 21, 21, 21)
# day variations
assert date_filter.parse('today', relative_base=test_now) == (datetime(1984, 4, 1, 0, 0, 0), datetime(1984, 4, 2, 0, 0, 0))
assert date_filter.parse('tomorrow', relative_base=test_now) == (datetime(1984, 4, 2, 0, 0, 0), datetime(1984, 4, 3, 0, 0, 0))
assert date_filter.parse('yesterday', relative_base=test_now) == (datetime(1984, 3, 31, 0, 0, 0), datetime(1984, 4, 1, 0, 0, 0))
assert date_filter.parse('5 days ago', relative_base=test_now) == (datetime(1984, 3, 27, 0, 0, 0), datetime(1984, 3, 28, 0, 0, 0))
# week variations
assert date_filter.parse('last week', relative_base=test_now) == (datetime(1984, 3, 18, 0, 0, 0), datetime(1984, 3, 25, 0, 0, 0))
assert date_filter.parse('2 weeks ago', relative_base=test_now) == (datetime(1984, 3, 11, 0, 0, 0), datetime(1984, 3, 18, 0, 0, 0))
# month variations
assert date_filter.parse('next month', relative_base=test_now) == (datetime(1984, 5, 1, 0, 0, 0), datetime(1984, 6, 1, 0, 0, 0))
assert date_filter.parse('2 months ago', relative_base=test_now) == (datetime(1984, 2, 1, 0, 0, 0), datetime(1984, 3, 1, 0, 0, 0))
# year variations
assert date_filter.parse('this year', relative_base=test_now) == (datetime(1984, 1, 1, 0, 0, 0), datetime(1985, 1, 1, 0, 0, 0))
assert date_filter.parse('20 years later', relative_base=test_now) == (datetime(2004, 1, 1, 0, 0, 0), datetime(2005, 1, 1, 0, 0, 0))
def test_date_filter_regex():
dtrange_match = re.search(date_filter.date_range_regex, 'head dt>"today" dt:"2020-01-01" tail')
assert dtrange_match.groups() == ('>', 'today', ':', '2020-01-01')
dtrange_match = re.search(date_filter.date_range_regex, 'head dt>="today" dt="2020-01-01"')
assert dtrange_match.groups() == ('>=', 'today', '=', '2020-01-01')
dtrange_match = re.search(date_filter.date_range_regex, 'head dt<"today" tail')
assert dtrange_match.groups() == ('<', 'today', None, None)
dtrange_match = re.search(date_filter.date_range_regex, 'head dt<="today"')
assert dtrange_match.groups() == ('<=', 'today', None, None)
dtrange_match = re.search(date_filter.date_range_regex, 'head tail')
assert dtrange_match.groups() == (None, None, None, None)