code: def test_random_contrast(magnitude):
      in_channels = 3
      in_height = 224
      in_width = 224
      x = torch.ones(
          size=(1, in_channels, in_width, in_height), dtype=torch.float
      )
  
      aug_out = random_contrast(
          x, magnitude=torch.tensor(magnitude, dtype=torch.float)
      )
  
      torch.testing.assert_allclose(
          actual=x.shape, expected=aug_out.shape
      )
  
imports: import sys
  import pytest
  from cvnets.neural_augmentor.utils.neural_aug_utils import *
label_desc: Use image node_id 3 for calling a global function with example usage: test_random_contrast(magnitude) without return types