def test_random_block_mask_image__aspect_ratio_one() -> None:
    """With aspect ratio 1.0 and num_mask=min_num_masks_per_block we expect a single,
    square masked block."""
    mask = utils.random_block_mask_image(
        size=(14, 14),
        num_masks=9,
        min_num_masks_per_block=9,
        min_block_aspect_ratio=1.0,
        max_block_aspect_ratio=1.0,
    )
    assert mask.sum(dim=0).max() == 3
    assert mask.sum(dim=1).max() == 3
With aspect ratio 1.0 and num_mask=min_num_masks_per_block we expect a single,
    square masked block.
tests/models/test_ModelUtils.py