Add sandbox support

Author: addisonlynch

pandas_datareader/iex/daily.py

@@ -53,6 +53,11 @@ def __init__(self, symbols=None, start=None, end=None, retry_count=3,
             raise ValueError('The IEX Cloud API key must be provided either '
                              'through the api_key variable or through the '
                              ' environment variable IEX_API_KEY')
+        # Support for sandbox environment (testing purposes)
+        if os.getenv("IEX_SANDBOX") == "enable":
+            self.sandbox = True
+        else:
+            self.sandbox = False
         self.api_key = api_key
         super(IEXDailyReader, self).__init__(symbols=symbols, start=start,
                                              end=end, retry_count=retry_count,
@@ -62,7 +67,10 @@ def __init__(self, symbols=None, start=None, end=None, retry_count=3,
     @property
     def url(self):
         """API URL"""
-        return 'https://sandbox.iexapis.com/stable/stock/market/batch'
+        if self.sandbox is True:
+            return 'https://sandbox.iexapis.com/stable/stock/market/batch'
+        else:
+            return 'https://cloud.iexapis.com/stable/stock/market/batch'
 
     @property
     def endpoint(self):

pandas_datareader/tests/test_iex_daily.py

@@ -1,68 +1,68 @@
-from datetime import datetime
-import os
-
-from pandas import DataFrame, MultiIndex
-
-import pytest
-
-import pandas_datareader.data as web
-
-
-@pytest.mark.skipif(os.getenv("IEX_SANDBOX") != 'enable',
-                    reason='All tests must be run in sandbox mode')
-class TestIEXDaily(object):
-
-    @classmethod
-    def setup_class(cls):
-        pytest.importorskip("lxml")
-
-    @property
-    def start(self):
-        return datetime(2015, 2, 9)
-
-    @property
-    def end(self):
-        return datetime(2017, 5, 24)
-
-    def test_iex_bad_symbol(self):
-        with pytest.raises(Exception):
-            web.DataReader("BADTICKER", "iex,", self.start, self.end)
-
-    def test_iex_bad_symbol_list(self):
-        with pytest.raises(Exception):
-            web.DataReader(["AAPL", "BADTICKER"], "iex",
-                           self.start, self.end)
-
-    def test_daily_invalid_date(self):
-        start = datetime(2000, 1, 5)
-        end = datetime(2017, 5, 24)
-        with pytest.raises(Exception):
-            web.DataReader(["AAPL", "TSLA"], "iex", start, end)
-
-    def test_single_symbol(self):
-        df = web.DataReader("AAPL", "iex", self.start, self.end)
-        assert list(df) == ["open", "high", "low", "close", "volume"]
-        assert len(df) == 578
-
-    def test_multiple_symbols(self):
-        syms = ["AAPL", "MSFT", "TSLA"]
-        df = web.DataReader(syms, "iex", self.start, self.end)
-        assert sorted(list(df.columns.levels[1])) == syms
-        for sym in syms:
-            assert len(df.xs(sym, level='Symbols', axis=1) == 578)
-
-    def test_multiple_symbols_2(self):
-        syms = ["AAPL", "MSFT", "TSLA"]
-        good_start = datetime(2017, 2, 9)
-        good_end = datetime(2017, 5, 24)
-        df = web.DataReader(syms, "iex", good_start, good_end)
-        assert isinstance(df, DataFrame)
-        assert isinstance(df.columns, MultiIndex)
-        assert len(df.columns.levels[1]) == 3
-        assert sorted(list(df.columns.levels[1])) == syms
-
-        a = df.xs("AAPL", axis=1, level='Symbols')
-        t = df.xs("TSLA", axis=1, level='Symbols')
-
-        assert len(a) == 73
-        assert len(t) == 73
+from datetime import datetime
+import os
+
+from pandas import DataFrame, MultiIndex
+
+import pytest
+
+import pandas_datareader.data as web
+
+
+@pytest.mark.skipif(os.getenv("IEX_SANDBOX") != 'enable',
+                    reason='All tests must be run in sandbox mode')
+class TestIEXDaily(object):
+
+    @classmethod
+    def setup_class(cls):
+        pytest.importorskip("lxml")
+
+    @property
+    def start(self):
+        return datetime(2015, 2, 9)
+
+    @property
+    def end(self):
+        return datetime(2017, 5, 24)
+
+    def test_iex_bad_symbol(self):
+        with pytest.raises(Exception):
+            web.DataReader("BADTICKER", "iex,", self.start, self.end)
+
+    def test_iex_bad_symbol_list(self):
+        with pytest.raises(Exception):
+            web.DataReader(["AAPL", "BADTICKER"], "iex",
+                           self.start, self.end)
+
+    def test_daily_invalid_date(self):
+        start = datetime(2000, 1, 5)
+        end = datetime(2017, 5, 24)
+        with pytest.raises(Exception):
+            web.DataReader(["AAPL", "TSLA"], "iex", start, end)
+
+    def test_single_symbol(self):
+        df = web.DataReader("AAPL", "iex", self.start, self.end)
+        assert list(df) == ["open", "high", "low", "close", "volume"]
+        assert len(df) == 578
+
+    def test_multiple_symbols(self):
+        syms = ["AAPL", "MSFT", "TSLA"]
+        df = web.DataReader(syms, "iex", self.start, self.end)
+        assert sorted(list(df.columns.levels[1])) == syms
+        for sym in syms:
+            assert len(df.xs(sym, level='Symbols', axis=1) == 578)
+
+    def test_multiple_symbols_2(self):
+        syms = ["AAPL", "MSFT", "TSLA"]
+        good_start = datetime(2017, 2, 9)
+        good_end = datetime(2017, 5, 24)
+        df = web.DataReader(syms, "iex", good_start, good_end)
+        assert isinstance(df, DataFrame)
+        assert isinstance(df.columns, MultiIndex)
+        assert len(df.columns.levels[1]) == 3
+        assert sorted(list(df.columns.levels[1])) == syms
+
+        a = df.xs("AAPL", axis=1, level='Symbols')
+        t = df.xs("TSLA", axis=1, level='Symbols')
+
+        assert len(a) == 73
+        assert len(t) == 73