diff --git a/src/bonsai/bonsai/core/tool.py b/src/bonsai/bonsai/core/tool.py index f71183910c..d4eaf402a8 100644 --- a/src/bonsai/bonsai/core/tool.py +++ b/src/bonsai/bonsai/core/tool.py @@ -933,7 +933,7 @@ class Spatial: def filter_products(cls, products, action): pass def get_active_container(cls): pass def get_container(cls, element): pass - def get_decomposed_elements(cls, container): pass + def get_decomposed_elements(cls, container, recursive): pass def get_object_matrix(cls, obj): pass def get_relative_object_matrix(cls, target_obj, relative_to_obj): pass def get_selected_product_types(cls): pass diff --git a/src/bonsai/test/core/test_spatial.py b/src/bonsai/test/core/test_spatial.py index 7d3679209a..a2b824fc91 100644 --- a/src/bonsai/test/core/test_spatial.py +++ b/src/bonsai/test/core/test_spatial.py @@ -115,6 +115,6 @@ class TestSelectSimilarContainer: def test_run(self, ifc, spatial): ifc.get_entity("obj").should_be_called().will_return("element") spatial.get_container("element").should_be_called().will_return("container") - spatial.get_decomposed_elements("container").should_be_called().will_return(["contained_element"]) + spatial.get_decomposed_elements("container", True).should_be_called().will_return(["contained_element"]) spatial.select_products(["contained_element"]).should_be_called() subject.select_similar_container(ifc, spatial, obj="obj")