Added colorscale test

Author: merenlin

plotly/tests/test_optional/test_opt_tracefactory.py

@@ -194,17 +194,19 @@ def test_dendrogram_random_matrix(self):
     def test_dendrogram_orientation(self):
         X = np.random.rand(5, 5) 
 
-        dendro_left = tls.FigureFactory.create_dendrogram(X, orientation='left')
-
+        dendro_left = tls.FigureFactory.create_dendrogram(
+                       X, orientation='left')
         self.assertEqual(len(dendro_left['layout']['yaxis']['ticktext']), 5)
         tickvals_left = np.array(dendro_left['layout']['yaxis']['tickvals'])
         self.assertTrue((tickvals_left <= 0).all())
 
-        dendro_right = tls.FigureFactory.create_dendrogram(X, orientation='right')
+        dendro_right = tls.FigureFactory.create_dendrogram(
+                        X, orientation='right')
         tickvals_right = np.array(dendro_right['layout']['yaxis']['tickvals'])
         self.assertTrue((tickvals_right >= 0).all())
 
-        dendro_bottom = tls.FigureFactory.create_dendrogram(X, orientation='bottom')
+        dendro_bottom = tls.FigureFactory.create_dendrogram(
+                        X, orientation='bottom')
         self.assertEqual(len(dendro_bottom['layout']['xaxis']['ticktext']), 5)
         tickvals_bottom = np.array(dendro_bottom['layout']['xaxis']['tickvals'])
         self.assertTrue((tickvals_bottom >= 0).all())
@@ -213,4 +215,25 @@ def test_dendrogram_orientation(self):
         tickvals_top = np.array(dendro_top['layout']['xaxis']['tickvals'])
         self.assertTrue((tickvals_top <= 0).all())
 
-      
+    def test_dendrogram_orientation(self):
+        X = np.array([[1, 2, 3, 4],
+                      [1, 1, 3, 4],
+                      [1, 2, 1, 4],
+                      [1, 2, 3, 1]])
+        greyscale = [
+                'rgb(0,0,0)',        # black
+                'rgb(05,105,105)',   # dim grey
+                'rgb(128,128,128)',  # grey
+                'rgb(169,169,169)',  # dark grey
+                'rgb(192,192,192)',  # silver
+                'rgb(211,211,211)',  # light grey
+                'rgb(220,220,220)',  # gainsboro
+                'rgb(245,245,245)']  # white smoke
+
+        dendro = tls.FigureFactory.create_dendrogram(X, colorscale=greyscale)
+        self.assertEqual(dendro["data"][0]['marker']['color'],
+                         'rgb(128,128,128)')
+        self.assertEqual(dendro["data"][1]['marker']['color'],
+                         'rgb(128,128,128)')
+        self.assertEqual(dendro["data"][2]['marker']['color'],
+                         'rgb(0,0,0)')

plotly/tools.py

@@ -3012,14 +3012,14 @@ def get_color_dict(self, colorscale):
 
         if colorscale is None:
             colorscale = [
-                "rgb(0,116,217)",   # blue
-                "rgb(35,205,205)",  # cyan
-                "rgb(61,153,112)",  # green
-                "rgb(40,35,35)",    # black
-                "rgb(133,20,75)",   # magenta
-                "rgb(255,65,54)",   # red
-                "rgb(255,255,255)",   # white
-                "rgb(255,220,0)"]   # yellow
+                'rgb(0,116,217)',   # blue
+                'rgb(35,205,205)',  # cyan
+                'rgb(61,153,112)',  # green
+                'rgb(40,35,35)',    # black
+                'rgb(133,20,75)',   # magenta
+                'rgb(255,65,54)',   # red
+                'rgb(255,255,255)',   # white
+                'rgb(255,220,0)']   # yellow
 
         for i in range(len(default_colors.keys())):
             k = default_colors.keys()[i]