def METHOD_NAME(self) -> None:
    mask = (
        torch.tensor([[0, 1], [2, 3]]),
        torch.tensor([]),
        torch.tensor([[4, 5], [6, 100]]),
        torch.tensor([[0, 1], [2, 3]]),
    )
    assert _get_max_feature_index(mask) == 100