From 9321d475d392bf23b9230a2365a5de015880963c Mon Sep 17 00:00:00 2001 From: Kakodkar Date: Mon, 17 Nov 2025 18:50:12 -0600 Subject: [PATCH] attr test for storage --- tests/attr_test.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/tests/attr_test.py b/tests/attr_test.py index d032f716..aa7bc0a8 100644 --- a/tests/attr_test.py +++ b/tests/attr_test.py @@ -2,7 +2,7 @@ from operator import is_ import pytest -from energia import Model, Resource, Transport, Process +from energia import Model, Resource, Transport, Process, Storage from energia.library.aliases import default_aliases @@ -76,3 +76,18 @@ def test_error(m): with pytest.raises(TypeError): m.rrrr = Resource(capex=5) + + +def test_storage_attrs(): + # storage pulls some attrs from constituents: + + m = Model() + m.r = Resource() + m.s = Storage() + m.s(m.r) == 1 + + for attr in ["capacity", "setup", "dismantle"]: + assert getattr(m.s, attr) == getattr(m.resources[-1], "inv" + attr) + + for attr in ["inventory"]: + assert getattr(m.s, attr) == getattr(m.resources[-1], attr)