Example #1
0
 def test_empty(self):
     mock_atom_index = mock.Mock()
     mock_atom_index.query.return_value = []
     mock_grammar = mock.Mock()
     mock_grammar.get_atoms.return_value = []
     mock_grammar.get_phrase_patterns.return_value = []
     mock_grammar.get_query_patterns.return_value = []
     result = interpret(mock_atom_index, mock_grammar, 'This is the way?')
     self.assertEqual(result, [])
     mock_atom_index.query.assert_called_with('This is the way')
Example #2
0
def search():
	query = request.query.q
	queries = interpret(atom_index, grammar, query)
	results = []
	for query in queries:
		results.append({ 
			'url' : '/results?sql=' + urllib.quote(query.get_sql()) + '&english=' + urllib.quote(query.get_english()),
			'short' : query.get_english_highlighted(),
			'sql' : query.get_sql()
		})
	return { 'results' : results }
Example #3
0
    def test_who_won(self):
        user_input = 'athletes most golds 2012'
        user_intent = 'Which athletes won the most golds in 2012?'

        results = interpret(self.atom_index, grammar, user_input)
        self.assertNotEqual(results, [])

        english_results = map(lambda r: r.get_english(), results)
        self.assertTrue(user_intent in english_results)

        sql_results = map(lambda r: r.get_sql(), results)
        data_results = map(self.db.query, sql_results)
        print map(lambda dr: dr[1], data_results)
        self.assertTrue(('Manster Bones', 2) in map(lambda dr: dr[1], data_results))
Example #4
0
 def test_no_dupes(self):
     results = interpret(self.atom_index, grammar, 'Manster Bones')
     english_results = map(lambda q: q.get_english(), results)
     self.assertEqual(len(english_results), len(list(set(english_results))))
Example #5
0
 def test_what_sports(self):
     user_input = 'what sports were contested?'
     results = interpret(self.atom_index, grammar, user_input)
     self.assertNotEqual(results, [])
     data_results = self.db.query(results[0].get_sql())
Example #6
0
 def test_how_many(self):
     user_input = 'how many silvers did Manster Bones win'
     results = interpret(self.atom_index, grammar, user_input)
     self.assertNotEqual(results, [])
Example #7
0
 def test_empty(self):
     result = interpret(self.atom_index, grammar, '')
     self.assertEqual(result, [])