Skip to content

Commit a22ca7d

Browse files
committed
add test for audit resource
1 parent 04f94ed commit a22ca7d

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

pydra/engine/tests/test_audit.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -302,3 +302,15 @@ def TestFunc(a: int, b: float = 0.1):
302302
# commented out to speed up testing
303303
collect_messages(tmpdir / funky._checksum, message_path, ld_op="compact")
304304
assert (tmpdir / funky._checksum / "messages.jsonld").exists()
305+
306+
307+
def test_audit_resource():
308+
309+
@python.define(outputs=["out"])
310+
def AddVar(a: int, b: int) -> int:
311+
return a + b
312+
313+
task0 = AddVar(a=4, b=5)
314+
outputs = task0(audit_flags=AuditFlag.RESOURCE)
315+
316+
print(outputs)

0 commit comments

Comments
 (0)