From fb1fd801eff403cbd314ccd55642d53585c41ef4 Mon Sep 17 00:00:00 2001 From: asdf Date: Wed, 25 Dec 2019 00:03:21 +1100 Subject: [PATCH] Search now case insensitive --- data.py | 2 +- tests/data_test.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/data.py b/data.py index 552990e..f336c5b 100644 --- a/data.py +++ b/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 str(record)) + query = (record for record in self.link_data if keyword.lower() in str(record).lower()) if query: search_results: set = set() diff --git a/tests/data_test.py b/tests/data_test.py index 179906b..f9c5854 100644 --- a/tests/data_test.py +++ b/tests/data_test.py @@ -180,7 +180,8 @@ 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"], + [62, "poster9", "1576461370.5580268", "", "c", "c", "ssskeywordsubstring"], + [61, "poster0", "1576461370.5580268", "", "c", "c", "KEYWORD capital"], ] test_results = [ @@ -206,7 +207,8 @@ 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"), + (62, "poster9", "1576461370.5580268", "", "c", "c", "ssskeywordsubstring"), + (61, "poster0", "1576461370.5580268", "", "c", "c", "KEYWORD capital"), ] self.assertEqual(link_data.search("keyword"), test_results)