kitty/kitty_tests/search_query_parser.py
2023-01-09 16:47:42 +05:30

31 lines
913 B
Python

#!/usr/bin/env python
# License: GPLv3 Copyright: 2022, Kovid Goyal <kovid at kovidgoyal.net>
from . import BaseTest
class TestSQP(BaseTest):
def test_search_query_parser(self):
from kitty.search_query_parser import ParseException, search
locations = 'id'
universal_set = {1, 2, 3, 4, 5}
def get_matches(location, query, candidates):
return {x for x in candidates if query == str(x)}
def t(q, expected=set()):
actual = search(q, locations, universal_set, get_matches)
self.ae(actual, expected)
t('id:1', {1})
t('id:"1"', {1})
t('id:1 and id:1', {1})
t('id:1 or id:2', {1, 2})
t('id:1 and id:2')
t('not id:1', universal_set - {1})
t('(id:1 or id:2) and id:1', {1})
self.assertRaises(ParseException, t, '1')
self.assertRaises(ParseException, t, '"id:1"')