forked from Ponysearch/Ponysearch
193 lines
8 KiB
Python
193 lines
8 KiB
Python
# -*- coding: utf-8 -*-
|
||
from collections import defaultdict
|
||
import mock
|
||
from searx.engines import wolframalpha_noapi
|
||
from searx.testing import SearxTestCase
|
||
|
||
|
||
class TestWolframAlphaNoAPIEngine(SearxTestCase):
|
||
|
||
def test_request(self):
|
||
query = 'test_query'
|
||
dicto = defaultdict(dict)
|
||
dicto['pageno'] = 1
|
||
params = wolframalpha_noapi.request(query, dicto)
|
||
self.assertIn('url', params)
|
||
self.assertIn(query, params['url'])
|
||
self.assertIn('wolframalpha.com', params['url'])
|
||
|
||
def test_response(self):
|
||
self.assertRaises(AttributeError, wolframalpha_noapi.response, None)
|
||
self.assertRaises(AttributeError, wolframalpha_noapi.response, [])
|
||
self.assertRaises(AttributeError, wolframalpha_noapi.response, '')
|
||
self.assertRaises(AttributeError, wolframalpha_noapi.response, '[]')
|
||
|
||
html = """
|
||
<!DOCTYPE html>
|
||
<title> Parangaricutirimícuaro - Wolfram|Alpha</title>
|
||
<meta charset="utf-8" />
|
||
<body>
|
||
<div id="closest">
|
||
<p class="pfail">Wolfram|Alpha doesn't know how to interpret your input.</p>
|
||
<div id="dtips">
|
||
<div class="tip">
|
||
<span class="tip-title">Tip: </span>
|
||
Check your spelling, and use English
|
||
<span class="tip-extra"></span>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</body>
|
||
</html>
|
||
"""
|
||
# test failed query
|
||
response = mock.Mock(text=html)
|
||
self.assertEqual(wolframalpha_noapi.response(response), [])
|
||
|
||
html = """
|
||
<!DOCTYPE html>
|
||
<title> sqrt(-1) - Wolfram|Alpha</title>
|
||
<meta charset="utf-8" />
|
||
<body>
|
||
<script type="text/javascript">
|
||
try {
|
||
if (typeof context.jsonArray.popups.pod_0100 == "undefined" ) {
|
||
context.jsonArray.popups.pod_0100 = [];
|
||
}
|
||
context.jsonArray.popups.pod_0100.push( {"stringified": "sqrt(-1)","mInput": "","mOutput": ""});
|
||
} catch(e) { }
|
||
|
||
try {
|
||
if (typeof context.jsonArray.popups.pod_0200 == "undefined" ) {
|
||
context.jsonArray.popups.pod_0200 = [];
|
||
}
|
||
context.jsonArray.popups.pod_0200.push( {"stringified": "i","mInput": "","mOutput": ""});
|
||
} catch(e) { }
|
||
</script>
|
||
</body>
|
||
</html>
|
||
"""
|
||
# test plaintext
|
||
response = mock.Mock(text=html)
|
||
results = wolframalpha_noapi.response(response)
|
||
self.assertEqual(type(results), list)
|
||
self.assertEqual(len(results), 2)
|
||
self.assertEquals('i', results[0]['answer'])
|
||
self.assertIn('sqrt(-1) - Wolfram|Alpha', results[1]['title'])
|
||
self.assertEquals('http://www.wolframalpha.com/input/?i=+sqrt%28-1%29', results[1]['url'])
|
||
|
||
html = """
|
||
<!DOCTYPE html>
|
||
<title> integral 1/x - Wolfram|Alpha</title>
|
||
<meta charset="utf-8" />
|
||
<body>
|
||
<script type="text/javascript">
|
||
try {
|
||
if (typeof context.jsonArray.popups.pod_0100 == "undefined" ) {
|
||
context.jsonArray.popups.pod_0100 = [];
|
||
}
|
||
context.jsonArray.popups.pod_0100.push( {"stringified": "integral 1\/x dx = log(x)+constant"});
|
||
} catch(e) { }
|
||
</script>
|
||
</body>
|
||
</html>
|
||
"""
|
||
# test integral
|
||
response = mock.Mock(text=html)
|
||
results = wolframalpha_noapi.response(response)
|
||
self.assertEqual(type(results), list)
|
||
self.assertEqual(len(results), 2)
|
||
self.assertIn('log(x)+c', results[0]['answer'])
|
||
self.assertIn('integral 1/x - Wolfram|Alpha', results[1]['title'])
|
||
self.assertEquals('http://www.wolframalpha.com/input/?i=+integral+1%2Fx', results[1]['url'])
|
||
|
||
html = """
|
||
<!DOCTYPE html>
|
||
<title> ∫1/x x - Wolfram|Alpha</title>
|
||
<meta charset="utf-8" />
|
||
<body>
|
||
<script type="text/javascript">
|
||
try {
|
||
if (typeof context.jsonArray.popups.pod_0100 == "undefined" ) {
|
||
context.jsonArray.popups.pod_0100 = [];
|
||
}
|
||
context.jsonArray.popups.pod_0100.push( {"stringified": "integral 1\/x dx = log(x)+constant"});
|
||
} catch(e) { }
|
||
</script>
|
||
</body>
|
||
</html>
|
||
"""
|
||
# test input in mathematical notation
|
||
response = mock.Mock(text=html)
|
||
results = wolframalpha_noapi.response(response)
|
||
self.assertEqual(type(results), list)
|
||
self.assertEqual(len(results), 2)
|
||
self.assertIn('log(x)+c', results[0]['answer'])
|
||
self.assertIn('∫1/x x - Wolfram|Alpha'.decode('utf-8'), results[1]['title'])
|
||
self.assertEquals('http://www.wolframalpha.com/input/?i=+%E2%88%AB1%2Fx+%EF%9D%8Cx', results[1]['url'])
|
||
|
||
html = """
|
||
<!DOCTYPE html>
|
||
<title> 1 euro to yen - Wolfram|Alpha</title>
|
||
<meta charset="utf-8" />
|
||
<body>
|
||
<script type="text/javascript">
|
||
try {
|
||
if (typeof context.jsonArray.popups.pod_0100 == "undefined" ) {
|
||
context.jsonArray.popups.pod_0100 = [];
|
||
}
|
||
context.jsonArray.popups.pod_0100.push( {"stringified": "convert euro1 (euro) to Japanese yen"});
|
||
} catch(e) { }
|
||
|
||
try {
|
||
if (typeof context.jsonArray.popups.pod_0200 == "undefined" ) {
|
||
context.jsonArray.popups.pod_0200 = [];
|
||
}
|
||
context.jsonArray.popups.pod_0200.push( {"stringified": "¥130.5 (Japanese yen)"});
|
||
} catch(e) { }
|
||
</script>
|
||
</body>
|
||
</html>
|
||
"""
|
||
# test output with htmlentity
|
||
response = mock.Mock(text=html)
|
||
results = wolframalpha_noapi.response(response)
|
||
self.assertEqual(type(results), list)
|
||
self.assertEqual(len(results), 2)
|
||
self.assertIn('¥'.decode('utf-8'), results[0]['answer'])
|
||
self.assertIn('1 euro to yen - Wolfram|Alpha', results[1]['title'])
|
||
self.assertEquals('http://www.wolframalpha.com/input/?i=+1+euro+to+yen', results[1]['url'])
|
||
|
||
html = """
|
||
<!DOCTYPE html>
|
||
<title> distance from nairobi to kyoto in inches - Wolfram|Alpha</title>
|
||
<meta charset="utf-8" />
|
||
<body>
|
||
<script type="text/javascript">
|
||
try {
|
||
if (typeof context.jsonArray.popups.pod_0100 == "undefined" ) {
|
||
context.jsonArray.popups.pod_0100 = [];
|
||
}
|
||
[...].pod_0100.push( {"stringified": "convert distance | from | Nairobi, Kenya\nto | Kyoto, Japan to inches"});
|
||
} catch(e) { }
|
||
|
||
try {
|
||
if (typeof context.jsonArray.popups.pod_0200 == "undefined" ) {
|
||
context.jsonArray.popups.pod_0200 = [];
|
||
}
|
||
pod_0200.push({"stringified": "4.295×10^8 inches","mOutput": "Quantity[4.295×10^8,&quot;Inches&quot;]"});
|
||
|
||
} catch(e) { }
|
||
</script>
|
||
</body>
|
||
</html>
|
||
"""
|
||
# test output with utf-8 character
|
||
response = mock.Mock(text=html)
|
||
results = wolframalpha_noapi.response(response)
|
||
self.assertEqual(type(results), list)
|
||
self.assertEqual(len(results), 2)
|
||
self.assertIn('4.295×10^8 inches'.decode('utf-8'), results[0]['answer'])
|
||
self.assertIn('distance from nairobi to kyoto in inches - Wolfram|Alpha', results[1]['title'])
|
||
self.assertEquals('http://www.wolframalpha.com/input/?i=+distance+from+nairobi+to+kyoto+in+inches',
|
||
results[1]['url'])
|