logo

searx

My custom branche(s) on searx, a meta-search engine git clone https://anongit.hacktivis.me/git/searx.git/

test_wolframalpha_api.py (6976B)

    

  1. # -*- coding: utf-8 -*-
  2. from collections import defaultdict
  3. import mock
  4. from requests import Request
  5. from searx.engines import wolframalpha_api
  6. from searx.testing import SearxTestCase
  9. class TestWolframAlphaAPIEngine(SearxTestCase):
  11.     def test_request(self):
  12.         query = 'test_query'
  13.         dicto = defaultdict(dict)
  14.         params = wolframalpha_api.request(query, dicto)
  16.         # TODO: test api_key
  17.         self.assertIn('url', params)
  18.         self.assertIn('https://api.wolframalpha.com/v2/query?', params['url'])
  19.         self.assertIn(query, params['url'])
  20.         self.assertEqual('https://www.wolframalpha.com/input/?i=test_query', params['headers']['Referer'])
  22.     def test_replace_pua_chars(self):
  23.         self.assertEqual('i', wolframalpha_api.replace_pua_chars(u'\uf74e'))
  25.     def test_response(self):
  26.         self.assertRaises(AttributeError, wolframalpha_api.response, None)
  27.         self.assertRaises(AttributeError, wolframalpha_api.response, [])
  28.         self.assertRaises(AttributeError, wolframalpha_api.response, '')
  29.         self.assertRaises(AttributeError, wolframalpha_api.response, '[]')
  31.         referer_url = 'referer_url'
  32.         request = Request(headers={'Referer': referer_url})
  34.         # test failure
  35.         xml = '''<?xml version='1.0' encoding='UTF-8'?>
  36.         <queryresult success='false' error='false' />
  37.         '''
  38.         response = mock.Mock(text=xml.encode('utf-8'))
  39.         self.assertEqual(wolframalpha_api.response(response), [])
  41.         # test basic case
  42.         xml = b"""<?xml version='1.0' encoding='UTF-8'?>
  43.         <queryresult success='true'
  44.             error='false'
  45.             numpods='3'
  46.             datatypes='Math'
  47.             id='queryresult_id'
  48.             host='http://www4c.wolframalpha.com'
  49.             related='related_url'
  50.             version='2.6'>
  51.             <pod title='Input'
  52.                  scanner='Identity'
  53.                  id='Input'
  54.                  numsubpods='1'>
  55.                   <subpod title=''>
  56.                        <img src='input_img_src.gif'
  57.                            alt='input_img_alt'
  58.                            title='input_img_title' />
  59.                        <plaintext>input_plaintext</plaintext>
  60.                   </subpod>
  61.              </pod>
  62.              <pod title='Result'
  63.                  scanner='Simplification'
  64.                  id='Result'
  65.                  numsubpods='1'
  66.                  primary='true'>
  67.                   <subpod title=''>
  68.                        <img src='result_img_src.gif'
  69.                            alt='result_img_alt'
  70.                            title='result_img_title' />
  71.                        <plaintext>result_plaintext</plaintext>
  72.                   </subpod>
  73.              </pod>
  74.              <pod title='Manipulatives illustration'
  75.                  scanner='Arithmetic'
  76.                  id='Illustration'
  77.                  numsubpods='1'>
  78.                   <subpod title=''>
  79.                        <img src='illustration_img_src.gif'
  80.                            alt='illustration_img_alt' />
  81.                        <plaintext>illustration_plaintext</plaintext>
  82.                   </subpod>
  83.              </pod>
  84.         </queryresult>
  85.         """
  86.         response = mock.Mock(text=xml, request=request)
  87.         results = wolframalpha_api.response(response)
  88.         self.assertEqual(type(results), list)
  89.         self.assertEqual(len(results), 2)
  90.         self.assertEqual('input_plaintext', results[0]['infobox'])
  92.         self.assertEqual(len(results[0]['attributes']), 3)
  93.         self.assertEqual('Input', results[0]['attributes'][0]['label'])
  94.         self.assertEqual('input_plaintext', results[0]['attributes'][0]['value'])
  95.         self.assertEqual('Result', results[0]['attributes'][1]['label'])
  96.         self.assertEqual('result_plaintext', results[0]['attributes'][1]['value'])
  97.         self.assertEqual('Manipulatives illustration', results[0]['attributes'][2]['label'])
  98.         self.assertEqual('illustration_img_src.gif', results[0]['attributes'][2]['image']['src'])
  99.         self.assertEqual('illustration_img_alt', results[0]['attributes'][2]['image']['alt'])
  101.         self.assertEqual(len(results[0]['urls']), 1)
  103.         self.assertEqual(referer_url, results[0]['urls'][0]['url'])
  104.         self.assertEqual('Wolfram|Alpha', results[0]['urls'][0]['title'])
  105.         self.assertEqual(referer_url, results[1]['url'])
  106.         self.assertEqual('Wolfram|Alpha (input_plaintext)', results[1]['title'])
  107.         self.assertIn('result_plaintext', results[1]['content'])
  109.         # test calc
  110.         xml = b"""<?xml version='1.0' encoding='UTF-8'?>
  111.         <queryresult success='true'
  112.             error='false'
  113.             numpods='2'
  114.             datatypes=''
  115.             parsetimedout='false'
  116.             id='queryresult_id'
  117.             host='http://www5b.wolframalpha.com'
  118.             related='related_url'
  119.             version='2.6' >
  120.             <pod title='Indefinite integral'
  121.                 scanner='Integral'
  122.                 id='IndefiniteIntegral'
  123.                 error='false'
  124.                 numsubpods='1'
  125.                 primary='true'>
  126.                 <subpod title=''>
  127.                     <img src='integral_image.gif'
  128.                         alt='integral_img_alt'
  129.                         title='integral_img_title' />
  130.                     <plaintext>integral_plaintext</plaintext>
  131.                 </subpod>
  132.             </pod>
  133.             <pod title='Plot of the integral'
  134.                 scanner='Integral'
  135.                 id='Plot'
  136.                 error='false'
  137.                 numsubpods='1'>
  138.                 <subpod title=''>
  139.                     <img src='plot.gif'
  140.                         alt='plot_alt'
  141.                         title='' />
  142.                     <plaintext></plaintext>
  143.                 </subpod>
  144.             </pod>
  145.         </queryresult>
  146.         """
  147.         response = mock.Mock(text=xml, request=request)
  148.         results = wolframalpha_api.response(response)
  149.         self.assertEqual(type(results), list)
  150.         self.assertEqual(len(results), 2)
  151.         self.assertEqual('integral_plaintext', results[0]['infobox'])
  153.         self.assertEqual(len(results[0]['attributes']), 2)
  154.         self.assertEqual('Indefinite integral', results[0]['attributes'][0]['label'])
  155.         self.assertEqual('integral_plaintext', results[0]['attributes'][0]['value'])
  156.         self.assertEqual('Plot of the integral', results[0]['attributes'][1]['label'])
  157.         self.assertEqual('plot.gif', results[0]['attributes'][1]['image']['src'])
  158.         self.assertEqual('plot_alt', results[0]['attributes'][1]['image']['alt'])
  160.         self.assertEqual(len(results[0]['urls']), 1)
  162.         self.assertEqual(referer_url, results[0]['urls'][0]['url'])
  163.         self.assertEqual('Wolfram|Alpha', results[0]['urls'][0]['title'])
  164.         self.assertEqual(referer_url, results[1]['url'])
  165.         self.assertEqual('Wolfram|Alpha (integral_plaintext)', results[1]['title'])
  166.         self.assertIn('integral_plaintext', results[1]['content'])