Fix __repr__ pretty printer for graph objs and add tests

Author: jonmmease

Diff for: plotly/tests/test_core/test_graph_objs/test_repr.py

@@ -0,0 +1,113 @@
+from __future__ import absolute_import
+from unittest import TestCase, SkipTest
+import plotly.graph_objs as go
+
+
+class TestGraphObjRepr(TestCase):
+
+    def test_trace_repr(self):
+        N = 100
+        scatt = go.Scatter(
+            y=list(range(N)),
+            marker={'color': 'green',
+                    'opacity': [e / N for e in range(N)]})
+
+        expected = """\
+Scatter(**{
+    'marker': {'color': 'green',
+               'opacity': [0.0, 0.01, 0.02, 0.03, 0.04, 0.05, 0.06, 0.07, 0.08,
+                           0.09, 0.1, 0.11, 0.12, 0.13, 0.14, 0.15, 0.16, 0.17,
+                           0.18, 0.19, 0.2, 0.21, 0.22, 0.23, 0.24, 0.25, 0.26,
+                           0.27, 0.28, 0.29, 0.3, 0.31, 0.32, 0.33, 0.34, 0.35,
+                           0.36, 0.37, 0.38, 0.39, 0.4, 0.41, 0.42, 0.43, 0.44,
+                           0.45, 0.46, 0.47, 0.48, 0.49, 0.5, 0.51, 0.52, 0.53,
+                           0.54, 0.55, 0.56, 0.57, 0.58, 0.59, 0.6, 0.61, 0.62,
+                           0.63, 0.64, 0.65, 0.66, 0.67, 0.68, 0.69, 0.7, 0.71,
+                           0.72, 0.73, 0.74, 0.75, 0.76, 0.77, 0.78, 0.79, 0.8,
+                           0.81, 0.82, 0.83, 0.84, 0.85, 0.86, 0.87, 0.88, 0.89,
+                           0.9, 0.91, 0.92, 0.93, 0.94, 0.95, 0.96, 0.97, 0.98,
+                           0.99]},
+    'type': 'scatter',
+    'y': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19,
+          20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37,
+          38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55,
+          56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73,
+          74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91,
+          92, 93, 94, 95, 96, 97, 98, 99]
+})"""
+        assert scatt.__repr__() == expected
+
+    def test_trace_repr_elided(self):
+        N = 1000
+        scatt = go.Scatter(
+            y=list(range(N)),
+            marker={'color': 'green',
+                    'opacity': [e / N for e in range(N)]})
+
+        expected = """\
+Scatter(**{
+    'marker': {'color': 'green',
+               'opacity': [0.0, 0.001, 0.002, ..., 0.997, 0.998, 0.999]},
+    'type': 'scatter',
+    'y': [0, 1, 2, ..., 997, 998, 999]
+})"""
+        assert scatt.__repr__() == expected
+
+    def test_trace_repr_numpy(self):
+        try:
+            import numpy as np
+        except ImportError:
+            raise SkipTest("Numpy not installed")
+
+        N = 100
+        scatt = go.Scatter(
+            y=np.arange(N),
+            marker={'color': 'green',
+                    'opacity': np.arange(N) / N})
+
+        expected = """\
+Scatter(**{
+    'marker': {'color': 'green',
+               'opacity': array([0.  , 0.01, 0.02, 0.03, 0.04, 0.05, 0.06, 0.07, 0.08, 0.09, 0.1 , 0.11,
+                                 0.12, 0.13, 0.14, 0.15, 0.16, 0.17, 0.18, 0.19, 0.2 , 0.21, 0.22, 0.23,
+                                 0.24, 0.25, 0.26, 0.27, 0.28, 0.29, 0.3 , 0.31, 0.32, 0.33, 0.34, 0.35,
+                                 0.36, 0.37, 0.38, 0.39, 0.4 , 0.41, 0.42, 0.43, 0.44, 0.45, 0.46, 0.47,
+                                 0.48, 0.49, 0.5 , 0.51, 0.52, 0.53, 0.54, 0.55, 0.56, 0.57, 0.58, 0.59,
+                                 0.6 , 0.61, 0.62, 0.63, 0.64, 0.65, 0.66, 0.67, 0.68, 0.69, 0.7 , 0.71,
+                                 0.72, 0.73, 0.74, 0.75, 0.76, 0.77, 0.78, 0.79, 0.8 , 0.81, 0.82, 0.83,
+                                 0.84, 0.85, 0.86, 0.87, 0.88, 0.89, 0.9 , 0.91, 0.92, 0.93, 0.94, 0.95,
+                                 0.96, 0.97, 0.98, 0.99])},
+    'type': 'scatter',
+    'y': array([ 0.,  1.,  2.,  3.,  4.,  5.,  6.,  7.,  8.,  9., 10., 11., 12., 13.,
+                14., 15., 16., 17., 18., 19., 20., 21., 22., 23., 24., 25., 26., 27.,
+                28., 29., 30., 31., 32., 33., 34., 35., 36., 37., 38., 39., 40., 41.,
+                42., 43., 44., 45., 46., 47., 48., 49., 50., 51., 52., 53., 54., 55.,
+                56., 57., 58., 59., 60., 61., 62., 63., 64., 65., 66., 67., 68., 69.,
+                70., 71., 72., 73., 74., 75., 76., 77., 78., 79., 80., 81., 82., 83.,
+                84., 85., 86., 87., 88., 89., 90., 91., 92., 93., 94., 95., 96., 97.,
+                98., 99.])
+})"""
+
+        assert scatt.__repr__() == expected
+
+    def test_trace_repr_numpy_elided(self):
+        try:
+            import numpy as np
+        except ImportError:
+            raise SkipTest("Numpy not installed")
+
+        N = 1000
+        scatt = go.Scatter(
+            y=np.arange(N),
+            marker={'color': 'green',
+                    'opacity': np.arange(N) / N})
+
+        expected = """\
+Scatter(**{
+    'marker': {'color': 'green',
+               'opacity': array([0.   , 0.001, 0.002, ..., 0.997, 0.998, 0.999])},
+    'type': 'scatter',
+    'y': array([  0.,   1.,   2., ..., 997., 998., 999.])
+})"""
+
+        assert scatt.__repr__() == expected

Diff for: plotly/utils.py

@@ -645,7 +645,7 @@ class ElidedPrettyPrinter(PrettyPrinter):
     """
     def __init__(self, *args, **kwargs):
         self.threshold = kwargs.pop('threshold', 200)
-        PrettyPrinter().__init__(*args, **kwargs)
+        super(ElidedPrettyPrinter, self).__init__(*args, **kwargs)
 
     def _format(self, val, stream, indent, allowance, context, level):
         if ElidedWrapper.is_wrappable(val):
@@ -655,5 +655,5 @@ def _format(self, val, stream, indent, allowance, context, level):
             return self._format(
                 elided_val, stream, indent, allowance, context, level)
         else:
-            return PrettyPrinter()._format(
+            return super(ElidedPrettyPrinter, self)._format(
                 val, stream, indent, allowance, context, level)