fix update tests to new INFO

Author: etpinard

plotly/tests/test_core/test_graph_objs/test_consistency.py

@@ -10,17 +10,17 @@
 
 
 def test_info_keys_in_key_to_name():
-    for key in graph_objs.INFO:
+    for key in graph_objs.INFO.keys():
         class_name = graph_objs.KEY_TO_NAME[key]
 
 
 def test_names_in_name_to_key():
-    for key in graph_objs.INFO:
+    for key in graph_objs.INFO.keys():
         class_name = graph_objs.KEY_TO_NAME[key]
         key_name = graph_objs.NAME_TO_KEY[class_name]
 
 
 def test_names_in_name_to_class():
-    for key in graph_objs.INFO:
+    for key in graph_objs.INFO.keys():
         class_name = graph_objs.KEY_TO_NAME[key]
         _class = graph_objs._factory(class_name)

plotly/tests/test_core/test_graph_reference/test_key_types.py

@@ -12,11 +12,11 @@
 
 def test_style_exists():
     checks = True
-    for obj_key, obj in list(go.INFO.items()):
-        if obj_key not in ['plotlylist', 'data', 'annotations', 'plotlydict', 'plotlytrace', 'trace']:
-            for attr_key, attr in list(obj.items()):
-                if 'type' not in attr:
+    for obj, stuff in list(go.INFO.items()):
+        if obj not in ['plotlylist', 'data', 'annotations', 'plotlydict', 'plotlytrace', 'trace']:
+            for attr_key, attr in list(stuff['keymeta'].items()):
+                if 'key_type' not in attr:
                     checks = False
                     print(obj_key, attr_key)
     if not checks:
-        raise Exception
+        raise Exception