diff --git a/nibabies/utils/bids.py b/nibabies/utils/bids.py index bfd63e76..010c977a 100644 --- a/nibabies/utils/bids.py +++ b/nibabies/utils/bids.py @@ -67,7 +67,9 @@ def __init__(self, bids_root: Path | str, spec: dict | Path | str | None = None, setattr(self, name, None) def __repr__(self): - return '\n'.join([name for name in self.names if getattr(self, name)]) + output = ["Derivatives |"] + output.extend([f" {attr}: {getattr(self, attr)}" for attr in self.names]) + return '\n'.join(output) def __contains__(self, val: str): return val in self.names diff --git a/nibabies/utils/tests/test_bids.py b/nibabies/utils/tests/test_bids.py index c46fe3e8..29ca735c 100644 --- a/nibabies/utils/tests/test_bids.py +++ b/nibabies/utils/tests/test_bids.py @@ -99,6 +99,7 @@ def test_derivatives( assert derivatives.aseg is None assert derivatives.t1w_aseg is None assert derivatives.t2w_aseg is None + assert "t1w_mask" in repr(derivatives) derivatives.populate(deriv_dir, subject_id='01') if mask: @@ -111,3 +112,5 @@ def test_derivatives( assert derivatives.references[aseg] else: assert derivatives.aseg == None + if t1w_mask: + assert "sub-01_space-T1w_desc-brain_mask.nii.gz" in repr(derivatives)