Search now matches if the keyword is a substring of any element
This commit is contained in:
parent
2e05eb3408
commit
5fdb0f4863
2
data.py
2
data.py
|
@ -190,7 +190,7 @@ class LinkData:
|
|||
if keyword == "":
|
||||
raise ValueError("a search keyword must be specified")
|
||||
|
||||
query = (record for record in self.link_data if keyword in record)
|
||||
query = (record for record in self.link_data if keyword in str(record))
|
||||
|
||||
if query:
|
||||
search_results: set = set()
|
||||
|
|
|
@ -180,7 +180,7 @@ class TestLinkDataSearch(unittest.TestCase):
|
|||
[65, "poster6", "1576461367.5580268", "", "keyword", "c", "c"],
|
||||
[64, "poster7", "1576461368.5580268", "", "c", "keyword", "c"],
|
||||
[63, "poster8", "1576461369.5580268", "", "c", "c", "keyword"],
|
||||
[63, "poster9", "1576461370.5580268", "", "c", "c", "z8keyworddui3"],
|
||||
[62, "poster9", "1576461370.5580268", "", "c", "c", "z8keyworddui3"],
|
||||
]
|
||||
|
||||
test_results = [
|
||||
|
@ -206,6 +206,7 @@ class TestLinkDataSearch(unittest.TestCase):
|
|||
(65, "poster6", "1576461367.5580268", "", "keyword", "c", "c"),
|
||||
(64, "poster7", "1576461368.5580268", "", "c", "keyword", "c"),
|
||||
(63, "poster8", "1576461369.5580268", "", "c", "c", "keyword"),
|
||||
(62, "poster9", "1576461370.5580268", "", "c", "c", "z8keyworddui3"),
|
||||
]
|
||||
|
||||
self.assertEqual(link_data.search("keyword"), test_results)
|
||||
|
|
|
@ -22,10 +22,10 @@ class TestPrintSearchResults(unittest.TestCase):
|
|||
call(
|
||||
"\nShowing results for keyword\n\n ID# DATE AUTHOR DESC "
|
||||
),
|
||||
call(" 1 2019-12-16 keyword c \n"),
|
||||
call(" 2 2019-12-16 poster6 c \n"),
|
||||
call(" 3 2019-12-16 poster7 c \n"),
|
||||
call(" 4 2019-12-16 poster8 keyword \n"),
|
||||
call(" 1 2019-12-16 keyword c "),
|
||||
call(" 2 2019-12-16 poster6 c "),
|
||||
call(" 3 2019-12-16 poster7 c "),
|
||||
call(" 4 2019-12-16 poster8 keyword "),
|
||||
]
|
||||
|
||||
linkulator.print_search_results(test_keyword, test_search_results)
|
Loading…
Reference in New Issue