code: def test_correct_args():
      """That that arguments have the right type and sizes."""
      with pytest.raises(
          ValueError, match="Expected argument `reduction`.*"
      ):
          _ = TotalVariation(reduction="diff")
  
      with pytest.raises(
          RuntimeError, match="Expected input `img` to.*"
      ):
          _ = total_variation(torch.randn(1, 2, 3))
  
imports: from functools import partial
  from typing import Any
  import pytest
  import torch
  from kornia.losses import total_variation
  from torchmetrics.functional.image.tv import total_variation
  from torchmetrics.image.tv import TotalVariation
  from unittests import _Input
  from unittests.helpers import seed_all
  from unittests.helpers.testers import MetricTester
label_desc: Use image node_id 3 for calling a global function with example usage: test_correct_args() without return types