Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add a test for snippets

Browse Source
This commit is contained in:
Kovid Goyal 2021-06-17 08:31:16 +05:30
parent 53b8bed17a
commit 06d34a2df9
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/calibre/db/tests/fts.py
View File

@ -38,6 +38,14 @@ CREATE VIRTUAL TABLE fts_row USING fts5vocab(fts_table, row);
def term_row_counts(self):
return dict(self.execute('SELECT term,doc FROM fts_row'))
def search(self, query, highlight_start='>', highlight_end='<', snippet_size=4):
snippet_size=max(1, min(snippet_size, 64))
stmt = (
f'SELECT snippet(fts_table, 0, "{highlight_start}", "{highlight_end}", "…", {snippet_size})'
' FROM fts_table WHERE fts_table MATCH ? ORDER BY RANK'
)
return list(self.execute(stmt, (query,)))
class FTSTest(BaseTest):
ae = BaseTest.assertEqual
@ -46,6 +54,8 @@ class FTSTest(BaseTest):
conn = TestConn()
conn.insert_text('two words, and a period. With another.')
conn.insert_text('and another re-init')
self.ae(conn.search("another"), [('and >another< re-init',), ('…With >another<.',)])
self.ae(conn.search("period"), [('…a >period<. With another.',)])
self.ae(conn.term_row_counts(), {'a': 1, 're': 1, 'init': 1, 'and': 2, 'another': 2, 'period': 1, 'two': 1, 'with': 1, 'words': 1})
conn = TestConn()
conn.insert_text('coộl')