BUG: Creating Index with the names argument

Author: spacesphere

pandas/core/indexes/base.py

@@ -112,14 +112,18 @@ class Index(IndexOpsMixin, PandasObject):
     dtype : NumPy dtype (default: object)
     copy : bool
         Make a copy of input ndarray
-    name : object
+    name : object, optional
         Name to be stored in the index
+    names : sequence of objects, optional
+        Names for the index levels
     tupleize_cols : bool (default: True)
         When True, attempt to create a MultiIndex if possible
 
     Notes
     -----
-    An Index instance can **only** contain hashable objects
+    An Index instance can **only** contain hashable objects.
+
+    Only one of `name` and `names` can be specified at the same time.
 
     Examples
     --------
@@ -177,10 +181,21 @@ class Index(IndexOpsMixin, PandasObject):
     str = accessor.AccessorProperty(strings.StringMethods)
 
     def __new__(cls, data=None, dtype=None, copy=False, name=None,
-                fastpath=False, tupleize_cols=True, **kwargs):
+                fastpath=False, tupleize_cols=True, names=None,
+                **kwargs):
+
+        if names is not None and name is not None:
+            raise TypeError("Can provide only one of names and name arguments")
 
-        if name is None and hasattr(data, 'name'):
-            name = data.name
+        if names is not None and not is_list_like(names):
+            raise TypeError("names must be list-like")
+
+        if name is None:
+            if hasattr(data, 'name'):
+                name = data.name
+            # extract `name` from `names` in case MultiIndex cannot be created
+            elif names:
+                name = names[0]
 
         if fastpath:
             return cls._simple_new(data, name)
@@ -359,8 +374,7 @@ def __new__(cls, data=None, dtype=None, copy=False, name=None,
                 # 10697
                 if all(isinstance(e, tuple) for e in data):
                     from .multi import MultiIndex
-                    return MultiIndex.from_tuples(
-                        data, names=name or kwargs.get('names'))
+                    return MultiIndex.from_tuples(data, names=names or name)
             # other iterable of some kind
             subarr = _asarray_tuplesafe(data, dtype=object)
             return Index(subarr, dtype=dtype, copy=copy, name=name, **kwargs)

pandas/tests/indexes/test_base.py

@@ -305,6 +305,30 @@ def test_constructor_simple_new(self):
         result = idx._simple_new(idx, 'obj')
         tm.assert_index_equal(result, idx)
 
+    def test_constructor_names(self):
+        # test both `name` and `names` provided
+        with pytest.raises(TypeError, match='only one of names and name'):
+            idx = Index([1, 2, 3], name='a', names=('a',))
+
+        # test non-list-like `names`
+        with pytest.raises(TypeError, match='names must be iterable'):
+            idx = Index([1, 2, 3], names='a')
+
+        # test using `name` for a flat `Index`
+        idx = Index([1, 2, 3], name='a')
+        assert idx.name == 'a'
+        assert idx.names == ('a',)
+
+        # test using `names` for a flat `Index`
+        idx = Index([1, 2, 3], names=('a',))
+        assert idx.name == 'a'
+        assert idx.names == ('a',)
+
+        # test using `names` for a `MultiIndex`
+        idx = Index([('A', 1), ('A', 2)], names=('a', 'b'))
+        assert idx.name is None
+        assert idx.names == ('a', 'b')
+
     def test_constructor_dtypes(self):
 
         for idx in [Index(np.array([1, 2, 3], dtype=int)),