move tests to generically tests for index

generify __unicode__ for Index

Author: jreback

pandas/core/index.py

@@ -410,15 +410,35 @@ def __unicode__(self):
         Invoked by unicode(df) in py2 only. Yields a Unicode String in both
         py2/py3.
         """
-        prepr = com.pprint_thing(self, escape_chars=('\t', '\r', '\n'),
-                                 quote_strings=True)
+        klass = self.__class__.__name__
+        space = ' ' * (len(klass) + 1)
+        data = self._format_data()
+        attrs = self._format_attrs()
+        prepr = (u(",\n%s") % space).join([u("%s=%s") % (k, v)
+                                          for k, v in attrs])
+        res = u("%s(%s,\n%s%s)") % (klass,
+                                    data,
+                                    space,
+                                    prepr)
+
+        return res
 
-        name_attr=''
-        if self.name:
-            name_attr = u(" %s=%s,") % ('name', self.name)
+    def _format_data(self):
+        """
+        Return the formatted data as a unicode string
+        """
+        return com.pprint_thing(self, escape_chars=('\t', '\r', '\n'),
+                                quote_strings=True)
 
-        return "%s(%s,%s dtype='%s')" % (type(self).__name__,
-                prepr, name_attr, self.dtype)
+    def _format_attrs(self):
+        """
+        Return a list of tuples of the (attr,formatted_value)
+        """
+        attrs = []
+        if self.name is not None:
+            attrs.append(('name',default_pprint(self.name)))
+        attrs.append(('dtype',"'%s'" % self.dtype))
+        return attrs
 
     def to_series(self, **kwargs):
         """

pandas/tests/test_index.py

@@ -118,6 +118,18 @@ def test_ndarray_compat_properties(self):
         idx.nbytes
         idx.values.nbytes
 
+    def test_repr_roundtrip(self):
+
+        idx = self.create_index()
+        tm.assert_index_equal(eval(repr(idx)),idx)
+
+    def test_str(self):
+
+        # test the string repr
+        idx = self.create_index()
+        idx.name = 'foo'
+        self.assertTrue("name=u'foo'" in str(idx))
+        self.assertTrue(idx.__class__.__name__ in str(idx))
 
 class TestIndex(Base, tm.TestCase):
     _holder = Index
@@ -2052,6 +2064,26 @@ def test_slice_keep_name(self):
 
 class DatetimeLike(Base):
 
+    def test_repr_roundtrip(self):
+        raise nose.SkipTest("Short reprs are not supported repr for Datetimelike indexes")
+
+    def test_str(self):
+
+        # test the string repr
+        idx = self.create_index()
+        idx.name = 'foo'
+        self.assertTrue("length=%s" % len(idx) in str(idx))
+        self.assertTrue("name=u'foo'" in str(idx))
+        self.assertTrue(idx.__class__.__name__ in str(idx))
+
+        if hasattr(idx,'tz'):
+            if idx.tz is not None:
+                self.assertTrue("tz='%s'" % idx.tz in str(idx))
+            else:
+                self.assertTrue("tz=None" in str(idx))
+        if hasattr(idx,'freq'):
+            self.assertTrue("freq='%s'" % idx.freqstr in str(idx))
+
     def test_view(self):
 
         i = self.create_index()
@@ -3951,8 +3983,9 @@ def test_repr_with_unicode_data(self):
             index = pd.DataFrame(d).set_index(["a", "b"]).index
             self.assertFalse("\\u" in repr(index))  # we don't want unicode-escaped
 
-    def test_repr_roundtrip(self):
-        tm.assert_index_equal(eval(repr(self.index)), self.index)
+    def test_str(self):
+        # tested elsewhere
+        pass
 
     def test_unicode_string_with_unicode(self):
         d = {"a": [u("\u05d0"), 2, 3], "b": [4, 5, 6], "c": [7, 8, 9]}

pandas/tseries/base.py

@@ -260,31 +260,48 @@ def _formatter_func(self):
         """
         return str
 
-    def _format_footer(self):
-        raise AbstractMethodError(self)
+    def _format_data(self):
+        """
+        Return the formatted data as a unicode string
+        """
 
-    def __unicode__(self):
         formatter = self._formatter_func
-        summary = str(self.__class__) + '\n'
-
         n = len(self)
         if n == 0:
-            pass
+            summary = ''
         elif n == 1:
             first = formatter(self[0])
-            summary += '[%s]\n' % first
+            summary = '[%s]' % first
         elif n == 2:
             first = formatter(self[0])
             last = formatter(self[-1])
-            summary += '[%s, %s]\n' % (first, last)
+            summary = '[%s, %s]' % (first, last)
         else:
             first = formatter(self[0])
             last = formatter(self[-1])
-            summary += '[%s, ..., %s]\n' % (first, last)
+            summary = '[%s, ..., %s]' % (first, last)
 
-        summary += self._format_footer()
         return summary
 
+    def _format_attrs(self):
+        """
+        Return a list of tuples of the (attr,formatted_value)
+        """
+        attrs = super(DatetimeIndexOpsMixin, self)._format_attrs()
+        attrs.append(('length',len(self)))
+        for attrib in self._attributes:
+            if attrib == 'freq':
+                freq = self.freqstr
+                if freq is not None:
+                    freq = "'%s'" % freq
+                attrs.append(('freq',freq))
+            elif attrib == 'tz':
+                tz = self.tz
+                if tz is not None:
+                    tz = "'%s'" % tz
+                attrs.append(('tz',tz))
+        return attrs
+
     @cache_readonly
     def _resolution(self):
         from pandas.tseries.frequencies import Resolution

pandas/tseries/index.py

@@ -683,10 +683,6 @@ def _format_native_types(self, na_rep=u('NaT'),
     def to_datetime(self, dayfirst=False):
         return self.copy()
 
-    def _format_footer(self):
-        tagline = 'Length: %d, Freq: %s, Timezone: %s'
-        return tagline % (len(self), self.freqstr, self.tz)
-
     def astype(self, dtype):
         dtype = np.dtype(dtype)
 

pandas/tseries/period.py

@@ -387,7 +387,7 @@ def to_datetime(self, dayfirst=False):
     qyear = _field_accessor('qyear', 1)
     days_in_month = _field_accessor('days_in_month', 11, "The number of days in the month")
     daysinmonth = days_in_month
-    
+
     def _get_object_array(self):
         freq = self.freq
         return np.array([ Period._from_ordinal(ordinal=x, freq=freq) for x in self.values], copy=False)
@@ -697,10 +697,6 @@ def __array_finalize__(self, obj):
         self.name = getattr(obj, 'name', None)
         self._reset_identity()
 
-    def _format_footer(self):
-        tagline = 'Length: %d, Freq: %s'
-        return tagline % (len(self), self.freqstr)
-
     def take(self, indices, axis=None):
         """
         Analogous to ndarray.take

pandas/tseries/tdi.py

@@ -274,10 +274,6 @@ def _formatter_func(self):
         from pandas.core.format import _get_format_timedelta64
         return _get_format_timedelta64(self, box=True)
 
-    def _format_footer(self):
-        tagline = 'Length: %d, Freq: %s'
-        return tagline % (len(self), self.freqstr)
-
     def __setstate__(self, state):
         """Necessary for making this object picklable"""
         if isinstance(state, dict):

pandas/util/testing.py

@@ -1633,7 +1633,7 @@ class _AssertRaisesContextmanager(object):
     def __init__(self, exception, regexp=None, *args, **kwargs):
         self.exception = exception
         if regexp is not None and not hasattr(regexp, "search"):
-            regexp = re.compile(regexp)
+            regexp = re.compile(regexp, re.DOTALL)
         self.regexp = regexp
 
     def __enter__(self):