def test__act_on_key_for_next_for_edge_case(self, mock_search,
mock__update_table):
assistant = LocalAssistant((), Option())
screen = Screen(assistant)
papers = [self.PAPER1]
screen._act_on_key(110, 2, 0, [], papers)
mock__update_table.assert_not_called()
def test__transform(self):
option = Option()
assistant = LocalAssistant(('keyword', ), option)
self.assertEqual(assistant._transform(option), ())
option = Option(year_low='2018')
assistant = LocalAssistant(('keyword', ), option)
self.assertEqual(assistant._transform(option), ('year:[2018 to]', ))
option = Option(year_high='2018')
assistant = LocalAssistant(('keyword', ), option)
self.assertEqual(assistant._transform(option), ('year:[to 2018]', ))
def test_search_previous_for_local_search(self, mock_method):
current_papers = [Paper('Awesome Study in 2018')]
previous_papers = [Paper('Awesome Study in 2017')]
mock_method.return_value = previous_papers
assistant = LocalAssistant((), Option(start=10))
returned_papers = assistant.search_previous(current_papers)
self.assertEqual(returned_papers, previous_papers)
assistant = LocalAssistant((), Option(start=0))
returned_papers = assistant.search_previous(current_papers)
self.assertEqual(returned_papers, current_papers)
def test__search(self):
paper = Paper('Awesome Study in 2018')
with patch.object(Index, 'search_every', return_value=[paper]):
option = Option(every=True)
assistant = LocalAssistant((), option)
self.assertEqual(assistant._search(), [paper])
with patch.object(Index, 'search', return_value=[paper]):
option = Option(start=0)
assistant = LocalAssistant((), option)
self.assertEqual(assistant._search(), [paper])
def kwic():
option = Option()
assistant = LocalAssistant((), option)
assistant.kwic()
return 0
def reload():
option = Option()
assistant = LocalAssistant((), option)
assistant.reload()
return 0
def list():
option = Option(start=0, every=True)
assistant = LocalAssistant((), option)
assistant.search()
return 0
def search(keywords, target, year_low, year_high):
option = Option(start=0, year_low=year_low, year_high=year_high)
assistant = LocalAssistant(keywords, option) if target == 'local' \
else GlobalAssistant(keywords, option)
assistant.search()
return 0
def test_reload(self, ngram_reload, index_reload, load):
assistant = LocalAssistant((), Option())
assistant.reload()
load.assert_called_once_with()
index_reload.assert_called_once_with({})
ngram_reload.assert_called_once_with({})
def test__act_on_key_for_next(self, mock_search, mock__update_table):
assistant = LocalAssistant((), Option())
screen = Screen(assistant)
papers = [self.PAPER1]
screen._act_on_key(110, 2, 0, [], papers) # 110: ord('n')
mock__update_table.assert_called_once_with([self.PAPER2])