Merge pull request #877 from bashtage/fix-quandl

MAINT: Correct bug in Quandl testing

Author: bashtage

pandas_datareader/tests/test_quandl.py

@@ -1,10 +1,12 @@
 import os
 
 import pytest
+import pandas as pd
 
 from pandas_datareader import data as web
 from pandas_datareader.compat import assert_frame_equal
 
+
 TEST_API_KEY = os.getenv("QUANDL_API_KEY")
 # Ensure blank TEST_API_KEY not used in pull request
 TEST_API_KEY = None if not TEST_API_KEY else TEST_API_KEY
@@ -51,7 +53,7 @@ def test_db_wiki_us(self):
                 "AdjVolume",
             ],
         )
-        assert df.Close.at[self.day10] == 7.70
+        assert df.Close.at[pd.to_datetime(self.day10)] == 7.70
 
     def test_db_fse_frankfurt(self):
         # ALV_X: Allianz SE
@@ -73,7 +75,7 @@ def test_db_fse_frankfurt(self):
                 "DailyTurnover",
             ],
         )
-        assert df.Close.at[self.day10] == 159.45
+        assert df.Close.at[pd.to_datetime(self.day10)] == 159.45
 
     def test_fse_eon(self):
         # EON_X: E.on Se
@@ -95,7 +97,7 @@ def test_fse_eon(self):
                 "High",
             ],
         )
-        assert df.Close.at[self.day2] == 14.03
+        assert df.Close.at[pd.to_datetime(self.day2)] == 14.03
 
     def test_db_euronext_be_fr_nl_pt(self):
         # FP: Total SA
@@ -105,7 +107,7 @@ def test_db_euronext_be_fr_nl_pt(self):
             "EURONEXT/FP", "quandl", self.start2, self.end2, api_key=TEST_API_KEY
         )
         self.check_headers(df, ["Open", "High", "Low", "Last", "Turnover", "Volume"])
-        assert df.Last.at[self.day2] == 42.525
+        assert df.Last.at[pd.to_datetime(self.day2)] == 42.525
         df2 = web.DataReader("FP.FR", "quandl", self.start2, self.end2)
         assert (df.Last == df2.Last).all()
 
@@ -132,7 +134,7 @@ def test_hk_hsbc_uk(self):
             ],
         )
         # as of 2017-06-11, Price == LastClose, all others are NaN
-        assert df.NominalPrice.at[self.day2] == 74.0
+        assert df.NominalPrice.at[pd.to_datetime(self.day2)] == 74.0
 
     def test_db_nse_in(self):
         # TCS: Tata Consutancy Services
@@ -151,15 +153,15 @@ def test_db_nse_in(self):
                 "TurnoverLacs",
             ],
         )
-        assert df.Close.at[self.day10] == 1259.05
+        assert df.Close.at[pd.to_datetime(self.day10)] == 1259.05
 
     def test_db_tse_jp(self):
         # TSE/6758: Sony Corp.
         df = web.DataReader(
             "TSE/6758", "quandl", self.start10, self.end10, api_key=TEST_API_KEY
         )
         self.check_headers(df, ["Open", "High", "Low", "Close", "Volume"])
-        assert df.Close.at[self.day10] == 5190.0
+        assert df.Close.at[pd.to_datetime(self.day10)] == 5190.0
 
         df2 = web.get_data_quandl(
             "TSE/6758", self.start10, self.end10, api_key=TEST_API_KEY
@@ -188,4 +190,4 @@ def test_db_hkex_cn(self):
                 "LotSize",
             ],
         )
-        assert df.High.at[self.day2] == 91.9
+        assert df.High.at[pd.to_datetime(self.day2)] == 91.9