瀏覽代碼

Merge pull request #553 from Maxqia/convert

[fix] currency_convert: make unit tests pass
Adam Tauber 9 年之前
父節點
當前提交
e9f56fe3ee
共有 2 個檔案被更改,包括 10 行新增13 行删除
  1. 3
    3
      searx/engines/currency_convert.py
  2. 7
    10
      tests/unit/engines/test_currency_convert.py

+ 3
- 3
searx/engines/currency_convert.py 查看文件

9
 url = 'https://download.finance.yahoo.com/d/quotes.csv?e=.csv&f=sl1d1t1&s={query}=X'
9
 url = 'https://download.finance.yahoo.com/d/quotes.csv?e=.csv&f=sl1d1t1&s={query}=X'
10
 weight = 100
10
 weight = 100
11
 
11
 
12
-parser_re = re.compile(u'.*?(\d+(?:\.\d+)?) ([^.0-9].+) (in|to) ([^\.]+)\W*$', re.I)  # noqa
12
+parser_re = re.compile(u'.*?(\d+(?:\.\d+)?) ([^.0-9]+) (?:in|to) ([^.0-9]+)', re.I)  # noqa
13
 
13
 
14
 db = 1
14
 db = 1
15
 
15
 
16
 
16
 
17
 def normalize_name(name):
17
 def normalize_name(name):
18
-    name = name.lower().replace('-', ' ')
18
+    name = name.lower().replace('-', ' ').rstrip('s')
19
     name = re.sub(' +', ' ', name)
19
     name = re.sub(' +', ' ', name)
20
     return unicodedata.normalize('NFKD', name).lower()
20
     return unicodedata.normalize('NFKD', name).lower()
21
 
21
 
40
         # wrong query
40
         # wrong query
41
         return params
41
         return params
42
 
42
 
43
-    ammount, from_currency, none, to_currency = m.groups()
43
+    ammount, from_currency, to_currency = m.groups()
44
     ammount = float(ammount)
44
     ammount = float(ammount)
45
     from_currency = name_to_iso4217(from_currency.strip())
45
     from_currency = name_to_iso4217(from_currency.strip())
46
     to_currency = name_to_iso4217(to_currency.strip())
46
     to_currency = name_to_iso4217(to_currency.strip())

+ 7
- 10
tests/unit/engines/test_currency_convert.py 查看文件

14
         params = currency_convert.request(query, dicto)
14
         params = currency_convert.request(query, dicto)
15
         self.assertNotIn('url', params)
15
         self.assertNotIn('url', params)
16
 
16
 
17
-        query = '1.1.1 EUR in USD'
18
-        params = currency_convert.request(query, dicto)
19
-        self.assertNotIn('url', params)
20
-
21
-        query = '10 eur in usd'
17
+        query = 'convert 10 Pound Sterlings to United States Dollars'
22
         params = currency_convert.request(query, dicto)
18
         params = currency_convert.request(query, dicto)
23
         self.assertIn('url', params)
19
         self.assertIn('url', params)
24
         self.assertIn('finance.yahoo.com', params['url'])
20
         self.assertIn('finance.yahoo.com', params['url'])
25
-        self.assertIn('EUR', params['url'])
21
+        self.assertIn('GBP', params['url'])
26
         self.assertIn('USD', params['url'])
22
         self.assertIn('USD', params['url'])
27
 
23
 
28
     def test_response(self):
24
     def test_response(self):
29
         dicto = defaultdict(dict)
25
         dicto = defaultdict(dict)
30
         dicto['ammount'] = float(10)
26
         dicto['ammount'] = float(10)
31
-        dicto['from'] = "EUR"
27
+        dicto['from'] = "GBP"
32
         dicto['to'] = "USD"
28
         dicto['to'] = "USD"
33
-        dicto['from_name'] = "euro"
29
+        dicto['from_name'] = "pound sterling"
34
         dicto['to_name'] = "United States dollar"
30
         dicto['to_name'] = "United States dollar"
35
         response = mock.Mock(text='a,b,c,d', search_params=dicto)
31
         response = mock.Mock(text='a,b,c,d', search_params=dicto)
36
         self.assertEqual(currency_convert.response(response), [])
32
         self.assertEqual(currency_convert.response(response), [])
40
         results = currency_convert.response(response)
36
         results = currency_convert.response(response)
41
         self.assertEqual(type(results), list)
37
         self.assertEqual(type(results), list)
42
         self.assertEqual(len(results), 1)
38
         self.assertEqual(len(results), 1)
43
-        self.assertEqual(results[0]['answer'], '10.0 EUR = 5.0 USD, 1 EUR (euro) = 0.5 USD (United States dollar)')
39
+        self.assertEqual(results[0]['answer'], '10.0 GBP = 5.0 USD, 1 GBP (pound sterling)' +
40
+                         ' = 0.5 USD (United States dollar)')
44
         now_date = datetime.now().strftime('%Y%m%d')
41
         now_date = datetime.now().strftime('%Y%m%d')
45
         self.assertEqual(results[0]['url'], 'https://finance.yahoo.com/currency/converter-results/' +
42
         self.assertEqual(results[0]['url'], 'https://finance.yahoo.com/currency/converter-results/' +
46
-                                            now_date + '/10.0-eur-to-usd.html')
43
+                                            now_date + '/10.0-gbp-to-usd.html')