fix tests for real

nicolaskruchten · nicolaskruchten · commit 868da9d39caf · 2020-07-08T15:29:46.000-04:00
diff --git a/packages/python/plotly/plotly/express/_core.py b/packages/python/plotly/plotly/express/_core.py
@@ -1351,7 +1351,7 @@ def build_dataframe(args, constructor):
                 )
 
     no_color = False
-    if type(args["color"]) == str and args["color"] == NO_COLOR:
+    if type(args.get("color", None)) == str and args["color"] == NO_COLOR:
         no_color = True
         args["color"] = None
     # now that things have been prepped, we do the systematic rewriting of `args`

Original file line number	Diff line number	Diff line change
`@@ -1351,7 +1351,7 @@ def build_dataframe(args, constructor):`
`1351`	`1351`	`)`
`1352`	`1352`
`1353`	`1353`	`no_color = False`
`1354`		`- if type(args["color"]) == str and args["color"] == NO_COLOR:`
	`1354`	`+ if type(args.get("color", None)) == str and args["color"] == NO_COLOR:`
`1355`	`1355`	`no_color = True`
`1356`	`1356`	`args["color"] = None`
`1357`	`1357`	# now that things have been prepped, we do the systematic rewriting of `args`