Skip to content

Commit 517a713

Browse files
committed
Add docstrings to test class and methods
1 parent e815c23 commit 517a713

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/transforms/test_rand_grid_distortiond.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,11 @@
7777

7878

7979
class TestRandGridDistortiond(unittest.TestCase):
80+
"""Test cases for RandGridDistortiond dictionary transform."""
81+
8082
@parameterized.expand(TESTS)
8183
def test_rand_grid_distortiond(self, input_param, seed, input_data, expected_val_img, expected_val_mask):
84+
"""Verify distortion produces expected output for image and mask keys."""
8285
g = RandGridDistortiond(**input_param)
8386
g.set_random_state(seed=seed)
8487
result = g(input_data)

0 commit comments

Comments
 (0)