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])