Set 'Date' column as index

Author: suminb

pandas_datareader/naver.py

@@ -15,6 +15,7 @@ class NaverDailyReader(_DailyBaseReader):
     :param interval: Not implemented
     :param adjust_dividends: Not implemented
     """
+
     def __init__(
         self,
         symbols=None,
@@ -80,9 +81,10 @@ def _read_one_data(self, url, params):
             parsed, columns=["Date", "Open", "High", "Low", "Close", "Volume"]
         )
         prices["Date"] = to_datetime(prices["Date"])
+        prices = prices.set_index("Date")
 
         # NOTE: See _get_params() for explanations.
-        return prices[(prices["Date"] >= self.start) & (prices["Date"] <= self.end)]
+        return prices[(prices.index >= self.start) & (prices.index <= self.end)]
 
     def _parse_xml_response(self, xml_content):
         """Parses XML response from the server.

pandas_datareader/tests/test_naver.py

@@ -18,9 +18,9 @@ def test_naver_daily_reader(self, symbol, start, end):
         end = datetime(*end)
         reader = DataReader(symbol, "naver", start, end)
 
-        assert reader.shape[1] == 6
-        assert reader["Date"].min() >= start
-        assert reader["Date"].max() <= end
+        assert reader.shape[1] == 5
+        assert reader.index.min() >= start
+        assert reader.index.max() <= end
 
     def test_bulk_fetch(self):
         with pytest.raises(NotImplementedError):