From 047c9e3ee9386bc96f7abb8b814f4902d9f0ecae Mon Sep 17 00:00:00 2001 From: Andrej730 Date: Thu, 1 Aug 2024 16:55:43 +0500 Subject: [PATCH] fix core aggregate module test after 6eb3c48 --- src/blenderbim/blenderbim/core/tool.py | 4 +++- src/blenderbim/test/core/test_aggregate.py | 1 + 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/blenderbim/blenderbim/core/tool.py b/src/blenderbim/blenderbim/core/tool.py index 261029e27f..41f9df6b8c 100644 --- a/src/blenderbim/blenderbim/core/tool.py +++ b/src/blenderbim/blenderbim/core/tool.py @@ -65,10 +65,12 @@ class Demo: @interface class Aggregate: - def can_aggregate(cls, relating_object, related_object): pass + def can_aggregate(cls, relating_obj, related_obj): pass + def has_physical_body_representation(cls, element): pass def disable_editing(cls, obj): pass def enable_editing(cls, obj): pass def get_container(cls, element): pass + def get_relating_object(cls, related_element): pass @interface diff --git a/src/blenderbim/test/core/test_aggregate.py b/src/blenderbim/test/core/test_aggregate.py index e4f3efe58d..81e967de94 100644 --- a/src/blenderbim/test/core/test_aggregate.py +++ b/src/blenderbim/test/core/test_aggregate.py @@ -36,6 +36,7 @@ class TestAssignObject: def test_run(self, ifc, aggregate, collector): aggregate.can_aggregate("relating_obj", "related_obj").should_be_called().will_return(True) ifc.get_entity("relating_obj").should_be_called().will_return("relating_object") + aggregate.has_physical_body_representation("relating_object").should_be_called().will_return(False) ifc.get_entity("related_obj").should_be_called().will_return("related_object") ifc.run( "aggregate.assign_object", products=["related_object"], relating_object="relating_object"