Update setup.py

Author: Hironsan

README.md

@@ -88,7 +88,7 @@ To download a pre-trained model, call `download` function:
 ```python
 >>> from anago.utils import download
 
->>> url = 'https://storage.googleapis.com/chakki/datasets/public/ner/models_en.zip'
+>>> url = 'https://storage.googleapis.com/chakki/datasets/public/ner/model_en.zip'
 >>> download(url)
 'Downloading...'
 'Complete!'

setup.py

@@ -23,12 +23,12 @@
     sys.exit()
 
 required = [
-    'Keras>=2.1.1', 'h5py>=2.7.1', 'scikit-learn>=0.19.1', 'numpy>=1.13.3', 'tensorflow>=1.4.0',
+    'Keras>=2.1.6', 'h5py>=2.7.1', 'scikit-learn>=0.19.1', 'numpy>=1.14.3', 'tensorflow>=1.8.0', 'requests>=2.18.4'
 ]
 
 setup(
     name=NAME,
-    version='1.0.1',
+    version='1.0.2',
     description=DESCRIPTION,
     long_description=long_description,
     author=AUTHOR,

tests/test_utils.py

@@ -1,7 +1,7 @@
 import os
 import unittest
 
-from anago.utils import load_data_and_labels, batch_iter, Vocabulary
+from anago.utils import load_data_and_labels, batch_iter, Vocabulary, download
 from anago.preprocessing import IndexTransformer
 
 
@@ -27,6 +27,18 @@ def test_batch_iter(self):
             y_gen.extend(y1)
         self.assertEqual(len(y_gen), len(y))
 
+    def test_download(self):
+        save_dir = os.path.join(os.path.dirname(__file__), 'models')
+        url = 'https://storage.googleapis.com/chakki/datasets/public/ner/model_en.zip'
+        download(url, save_dir)
+
+        weights_file = os.path.join(save_dir, 'weights.h5')
+        params_file = os.path.join(save_dir, 'params.json')
+        preprocessor_file = os.path.join(save_dir, 'preprocessor.pickle')
+        self.assertTrue(os.path.exists(weights_file))
+        self.assertTrue(os.path.exists(params_file))
+        self.assertTrue(os.path.exists(preprocessor_file))
+
 
 class TestVocabulary(unittest.TestCase):