TST: ADD test on Series index auto-completion

Author: Martin Journois

pandas/tests/series/test_api.py

@@ -10,7 +10,7 @@
 from pandas import Index, Series, DataFrame, date_range
 from pandas.core.indexes.datetimes import Timestamp
 
-from pandas.compat import range
+from pandas.compat import range, lzip, isidentifier, string_types
 from pandas import compat
 import pandas.io.formats.printing as printing
 from pandas.util.testing import (assert_series_equal,
@@ -234,6 +234,39 @@ def test_tab_completion(self):
         assert 'str' not in dir(s)
         assert 'dt' in dir(s)  # as it is a datetime categorical
 
+    def test_index_tab_completion(self):
+        # dir contains string-like values of the Index.
+        for index in [tm.makeUnicodeIndex(10),
+                      tm.makeStringIndex(100),
+                      tm.makeCategoricalIndex(100),
+                      Index(['foo', 'foo',
+                             'bar', 'bar',
+                             'baz', 'baz']),
+                      tm.makeDateIndex(100),
+                      tm.makePeriodIndex(100),
+                      tm.makeTimedeltaIndex(100),
+                      tm.makeIntIndex(100),
+                      tm.makeUIntIndex(100),
+                      tm.makeIntIndex(100),
+                      tm.makeFloatIndex(100),
+                      Index([True, False]),
+                      ]:
+            s = pd.Series(index=index)
+            dir_s = dir(s)
+            for x in s.index:
+                assert (not isinstance(x, string_types) or
+                        not isidentifier(x) or x in dir_s)
+
+        # dir contains string-like values of the MultiIndex first level.
+        for index in [pd.MultiIndex.from_tuples(lzip('ABCD', 'EFGH')),
+                      pd.MultiIndex.from_tuples(lzip([0, 1, 2, 3], 'EFGH')),
+                      ]:
+            s = pd.Series(index=index)
+            dir_s = dir(s)
+            for x in s.index:
+                assert (not isinstance(x[0], string_types) or
+                        not isidentifier(x[0]) or x[0] in dir_s)
+
     def test_not_hashable(self):
         s_empty = Series()
         s = Series([1])