Skip to content

Commit 17c0e23

Browse files
committed
Add missing docstrings to regression test
Signed-off-by: Harrisaint <hgmartin1116@gmail.com>
1 parent 8309c22 commit 17c0e23

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

tests/data/utils/test_compute_shape_offset.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,17 @@
2020

2121

2222
class TestComputeShapeOffsetRegression(unittest.TestCase):
23+
"""Regression tests for `compute_shape_offset` input-shape handling."""
24+
2325
def test_pytorch_size_input(self):
26+
"""Validate `torch.Size` input produces expected shape and offset.
27+
28+
Returns:
29+
None.
30+
31+
Raises:
32+
AssertionError: If computed shape/offset are not as expected.
33+
"""
2434
# 1. Create a PyTorch Size object (which triggered the original bug)
2535
spatial_shape = torch.Size([10, 10, 10])
2636
in_affine = np.eye(4)

0 commit comments

Comments
 (0)