Update EnigmaReader Tests

Author: hydrosquall

pandas_datareader/enigma.py

@@ -38,7 +38,7 @@ def __init__(self,
                  dataset_id=None,
                  api_key=None,
                  retry_count=5,
-                 pause=0.5,
+                 pause=.75,
                  session=None):
 
         super(EnigmaReader, self).__init__(symbols=[],
@@ -99,9 +99,9 @@ def _get(self, url):
                 else:
                     raise e
 
-    def get_current_snapshot_id(self):
+    def get_current_snapshot_id(self, dataset_id):
         """Get ID of the most current snapshot of a dataset"""
-        dataset_metadata = self.get_dataset_metadata()
+        dataset_metadata = self.get_dataset_metadata(dataset_id)
         return dataset_metadata['current_snapshot']['id']
 
     def get_dataset_metadata(self, dataset_id):

pandas_datareader/tests/test_enigma.py

@@ -11,32 +11,40 @@
 @pytest.mark.skipif(TEST_API_KEY is None, reason="no enigma_api_key")
 class TestEnigma(object):
 
+    @property
+    def dataset_id():
+        """
+        USDA Food Recall Archive
+        Selected for being a relatively small dataset.
+        https://public.enigma.com/datasets/292129b0-1275-44c8-a6a3-2a0881f24fe1
+        """
+        return "292129b0-1275-44c8-a6a3-2a0881f24fe1"
+
     @classmethod
     def setup_class(cls):
         pytest.importorskip("lxml")
 
     def test_enigma_datareader(self):
         try:
-            df = web.DataReader('enigma.inspections.restaurants.fl',
+            df = web.DataReader(self.dataset_id,
                                 'enigma', access_key=TEST_API_KEY)
-            assert 'serialid' in df.columns
+            assert 'case_number' in df.columns
         except HTTPError as e:
             pytest.skip(e)
 
     def test_enigma_get_data_enigma(self):
         try:
-            df = pdr.get_data_enigma(
-                'enigma.inspections.restaurants.fl', TEST_API_KEY)
-            assert 'serialid' in df.columns
+            df = pdr.get_data_enigma(self.dataset_id, TEST_API_KEY)
+            assert 'case_number' in df.columns
         except HTTPError as e:
             pytest.skip(e)
 
     def test_bad_key(self):
         with pytest.raises(HTTPError):
-            web.DataReader('enigma.inspections.restaurants.fl',
+            web.DataReader(self.dataset_id,
                            'enigma', access_key=TEST_API_KEY + 'xxx')
 
-    def test_bad_url(self):
+    def test_bad_dataset_id(self):
         with pytest.raises(HTTPError):
-            web.DataReader('enigma.inspections.restaurants.fllzzy',
+            web.DataReader('zzzzzzzz-zzzz-zzzz-zzzz-zzzzzzzzzzz',
                            'enigma', access_key=TEST_API_KEY)