Check that QUANDL_API_KEY is set

msarm · Jul 17, 2018 · 006c156 · 006c156
1 parent c37af0e
commit 006c156
Showing 1 changed file with 16 additions and 12 deletions.
diff --git a/testcases/quandl_test.py b/testcases/quandl_test.py
@@ -37,13 +37,17 @@
     pass
 
 
+QUANDL_API_KEY = os.getenv("QUANDL_API_KEY")
+assert QUANDL_API_KEY is not None, "QUANDL_API_KEY not set"
+
+
 class ToolsTestCase(common.TestCase):
 
     def testDownloadAndParseDailyUsingApiKey(self):
         with common.TmpDir() as tmpPath:
             instrument = "ORCL"
             path = os.path.join(tmpPath, "quandl-daily-orcl-2010.csv")
-            quandl.download_daily_bars("WIKI", instrument, 2010, path, authToken=os.getenv("QUANDL_API_KEY"))
+            quandl.download_daily_bars("WIKI", instrument, 2010, path, authToken=QUANDL_API_KEY)
             bf = quandlfeed.Feed()
             bf.addBarsFromCSV(instrument, path)
             bf.loadAll()
@@ -62,7 +66,7 @@ def testDownloadAndParseDaily_UseAdjClose(self):
         with common.TmpDir() as tmpPath:
             instrument = "ORCL"
             path = os.path.join(tmpPath, "quandl-daily-orcl-2010.csv")
-            quandl.download_daily_bars("WIKI", instrument, 2010, path, authToken=os.getenv("QUANDL_API_KEY"))
+            quandl.download_daily_bars("WIKI", instrument, 2010, path, authToken=QUANDL_API_KEY)
             bf = quandlfeed.Feed()
             bf.addBarsFromCSV(instrument, path)
             # Need to setUseAdjustedValues(True) after loading the file because we
@@ -85,7 +89,7 @@ def testDownloadAndParseDailyNoAdjClose(self):
             instrument = "IWG"
             year = 2017
             path = os.path.join(tmpPath, "quandl-daily-%s-%s.csv" % (instrument, year))
-            quandl.download_daily_bars("LSE", instrument, year, path, authToken=os.getenv("QUANDL_API_KEY"))
+            quandl.download_daily_bars("LSE", instrument, year, path, authToken=QUANDL_API_KEY)
             bf = quandlfeed.Feed()
             bf.setNoAdjClose()
             bf.setColumnName("open", "Price")
@@ -105,7 +109,7 @@ def testDownloadAndParseWeekly(self):
         with common.TmpDir() as tmpPath:
             instrument = "AAPL"
             path = os.path.join(tmpPath, "quandl-aapl-weekly-2010.csv")
-            quandl.download_weekly_bars("WIKI", instrument, 2010, path, authToken=os.getenv("QUANDL_API_KEY"))
+            quandl.download_weekly_bars("WIKI", instrument, 2010, path, authToken=QUANDL_API_KEY)
             bf = quandlfeed.Feed(frequency=bar.Frequency.WEEK)
             bf.addBarsFromCSV(instrument, path)
             bf.loadAll()
@@ -131,7 +135,7 @@ def testInvalidFrequency(self):
     def testBuildFeedDaily(self):
         with common.TmpDir() as tmpPath:
             instrument = "ORCL"
-            bf = quandl.build_feed("WIKI", [instrument], 2010, 2010, tmpPath, authToken=os.getenv("QUANDL_API_KEY"))
+            bf = quandl.build_feed("WIKI", [instrument], 2010, 2010, tmpPath, authToken=QUANDL_API_KEY)
             bf.loadAll()
             self.assertEquals(bf[instrument][-1].getDateTime(), datetime.datetime(2010, 12, 31))
             self.assertEquals(bf[instrument][-1].getOpen(), 31.22)
@@ -149,7 +153,7 @@ def testBuildFeedWeekly(self):
             instrument = "AAPL"
             bf = quandl.build_feed(
                 "WIKI", [instrument], 2010, 2010, tmpPath, bar.Frequency.WEEK,
-                authToken=os.getenv("QUANDL_API_KEY")
+                authToken=QUANDL_API_KEY
             )
             bf.loadAll()
             # Quandl used to report 2010-1-3 as the first week of 2010.
@@ -175,14 +179,14 @@ def testInvalidInstrument(self):
             with common.TmpDir() as tmpPath:
                 quandl.build_feed(
                     instrument, [instrument], 2010, 2010, tmpPath, bar.Frequency.WEEK,
-                    authToken=os.getenv("QUANDL_API_KEY")
+                    authToken=QUANDL_API_KEY
                 )
 
         # Skip errors.
         with common.TmpDir() as tmpPath:
             bf = quandl.build_feed(
                 instrument, [instrument], 2010, 2010, tmpPath, bar.Frequency.WEEK, skipErrors=True,
-                authToken=os.getenv("QUANDL_API_KEY")
+                authToken=QUANDL_API_KEY
             )
             bf.loadAll()
             self.assertNotIn(instrument, bf)
@@ -197,7 +201,7 @@ def testMapColumnNames(self):
             year = 2017
             bf = quandl.build_feed(
                 "LSE", [instrument], year, year, tmpPath, columnNames=column_names, skipMalformedBars=True,
-                authToken=os.getenv("QUANDL_API_KEY")
+                authToken=QUANDL_API_KEY
             )
             bf.setNoAdjClose()
             bf.loadAll()
@@ -218,7 +222,7 @@ def testExtraColumns(self):
             }
             bf = quandl.build_feed(
                 "BITSTAMP", ["USD"], 2014, 2014, tmpPath, columnNames=columnNames,
-                authToken=os.getenv("QUANDL_API_KEY")
+                authToken=QUANDL_API_KEY
             )
             bf.loadAll()
 
@@ -238,7 +242,7 @@ def testNoAdjClose(self):
             }
             bf = quandl.build_feed(
                 "BITSTAMP", ["USD"], 2014, 2014, tmpPath, columnNames=columnNames,
-                authToken=os.getenv("QUANDL_API_KEY")
+                authToken=QUANDL_API_KEY
             )
             bf.loadAll()
 
@@ -250,7 +254,7 @@ def testBuildFeedDailyCreatingDir(self):
         shutil.rmtree(tmpPath)
         try:
             instrument = "ORCL"
-            bf = quandl.build_feed("WIKI", [instrument], 2010, 2010, tmpPath, authToken=os.getenv("QUANDL_API_KEY"))
+            bf = quandl.build_feed("WIKI", [instrument], 2010, 2010, tmpPath, authToken=QUANDL_API_KEY)
             bf.loadAll()
 
             self.assertEquals(bf[instrument][-1].getDateTime(), datetime.datetime(2010, 12, 31))