From 051ddbe527e67890782643372fd9366c5577b449 Mon Sep 17 00:00:00 2001 From: Hanson-HSChang Date: Mon, 9 Sep 2024 15:34:09 -0500 Subject: [PATCH] wip: a quick fix to the tests for stack --- tests/geometry/stack/test_base_stack_stacking.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/tests/geometry/stack/test_base_stack_stacking.py b/tests/geometry/stack/test_base_stack_stacking.py index f0257c2..612d203 100644 --- a/tests/geometry/stack/test_base_stack_stacking.py +++ b/tests/geometry/stack/test_base_stack_stacking.py @@ -7,6 +7,11 @@ class MockObjectToStack: def __init__(self, value): self.value = value + self.material_properties = {} + + @property + def material(self): + return self.material_properties @property def object(self): @@ -19,6 +24,9 @@ def create(cls, states): def update_states(self, value): self.value = value + def update_material(self, **kwargs): + self.material_properties.update(kwargs) + class MockStack(BaseStack[MockObjectToStack]): DefaultType = MockObjectToStack @@ -93,6 +101,11 @@ def test_update_wrong_size(update_data): class MockObjectToStack2: def __init__(self, value1, value2): self.value = (value1, value2) + self.material_properties = {} + + @property + def material(self): + return self.material_properties @property def object(self): @@ -105,6 +118,9 @@ def create(cls, states): def update_states(self, value1, value2): self.value = (value1, value2) + def update_material(self, **kwargs): + self.material_properties.update(kwargs) + class Mock2Stack(BaseStack[MockObjectToStack2]): DefaultType = MockObjectToStack2