Test fixes

Author: jonmmease

packages/python/plotly/plotly/basedatatypes.py

@@ -3098,21 +3098,25 @@ def _get_child_props(self, child):
             return None
         else:
             # ### Child a compound property ###
-            if child.plotly_name in self._compound_props:
-                return self._props.get(child.plotly_name, None)
+            if child.plotly_name in self:
+                from _plotly_utils.basevalidators import CompoundValidator, CompoundArrayValidator
+                validator = self._get_validator(child.plotly_name)
 
-            # ### Child an element of a compound array property ###
-            elif child.plotly_name in self._compound_array_props:
-                children = self._compound_array_props[child.plotly_name]
-                child_ind = BaseFigure._index_is(children, child)
-                assert child_ind is not None
+                if isinstance(validator, CompoundValidator):
+                    return self._props.get(child.plotly_name, None)
 
-                children_props = self._props.get(child.plotly_name, None)
-                return (
-                    children_props[child_ind]
-                    if children_props is not None and len(children_props) > child_ind
-                    else None
-                )
+                # ### Child an element of a compound array property ###
+                elif isinstance(validator, CompoundArrayValidator):
+                    children = self[child.plotly_name]
+                    child_ind = BaseFigure._index_is(children, child)
+                    assert child_ind is not None
+
+                    children_props = self._props.get(child.plotly_name, None)
+                    return (
+                        children_props[child_ind]
+                        if children_props is not None and len(children_props) > child_ind
+                        else None
+                    )
 
             # ### Invalid child ###
             else:
@@ -3320,7 +3324,9 @@ def __getitem__(self, prop):
         -------
         Any
         """
-        from _plotly_utils.basevalidators import CompoundValidator, CompoundArrayValidator
+        from _plotly_utils.basevalidators import (
+            CompoundValidator, CompoundArrayValidator, BaseDataValidator
+        )
 
         # Normalize prop
         # --------------
@@ -3346,10 +3352,15 @@ def __getitem__(self, prop):
             if isinstance(validator, CompoundValidator):
                 if self._compound_props.get(prop, None) is None:
                     # Init compound objects
-                    self._compound_props[prop] = validator.data_class(_parent=self)
+                    self._compound_props[prop] = validator.data_class(
+                        _parent=self, plotly_name=prop
+                    )
+                    # Update plotly_name value in case the validator applies
+                    # non-standard name (e.g. imagedefaults instead of image)
+                    self._compound_props[prop]._plotly_name = prop
 
                 return validator.present(self._compound_props[prop])
-            elif isinstance(validator, CompoundArrayValidator):
+            elif isinstance(validator, (CompoundArrayValidator, BaseDataValidator)):
                 if self._compound_array_props.get(prop, None) is None:
                     # Init list of compound objects
                     self._compound_array_props[prop] = [