fix tests

nicolaskruchten · nicolaskruchten · commit 0dd2f39c868d · 2019-11-28T09:27:36.000-05:00
diff --git a/packages/python/plotly/plotly/express/_chart_types.py b/packages/python/plotly/plotly/express/_chart_types.py
@@ -1320,4 +1320,3 @@ def funnel_area(
 
 
 funnel_area.__doc__ = make_docstring(funnel_area)
-
diff --git a/packages/python/plotly/plotly/express/_core.py b/packages/python/plotly/plotly/express/_core.py
@@ -1045,18 +1045,17 @@ def infer_config(args, constructor, trace_patch):
                     attrs.append("color")
                     args["color_is_continuous"] = False
                 else:
-                    if constructor not in [go.Pie, go.Funnelarea]:
-                        grouped_attrs.append("marker.color")
+                    grouped_attrs.append("marker.color")
         elif "line_group" in args or constructor == go.Histogram2dContour:
             grouped_attrs.append("line.color")
-        elif constructor not in [go.Pie, go.Funnelarea]:
-            grouped_attrs.append("marker.color")
-        else:
+        elif constructor in [go.Pie, go.Funnelarea]:
             attrs.append("color")
-            if constructor in [go.Pie, go.Funnelarea]:
+            if args["color"]:
                 if args["hover_data"] is None:
                     args["hover_data"] = []
                 args["hover_data"].append(args["color"])
+        else:
+            grouped_attrs.append("marker.color")
 
         show_colorbar = bool(
             "color" in attrs

Original file line number	Diff line number	Diff line change
`@@ -1320,4 +1320,3 @@ def funnel_area(`
`1320`	`1320`
`1321`	`1321`
`1322`	`1322`	`funnel_area.__doc__ = make_docstring(funnel_area)`
`1323`		`-`