code: def test_shot_noise():
      a = torch.rand(4, 3, 100, 100)
      a1 = torchdrift.data.functional.shot_noise(a, severity=4)
      a2 = torchdrift.data.functional.gaussian_noise(a, severity=4)
      assert (a1 - a).abs().max() < (a2 - a).abs().max()
  
file_path: TorchDrift/test/test_corruption_functions.py
imports: import pytest
  import torchdrift
  import torch
  import scipy.ndimage
label_desc: Use image node_id 2 for calling a global function with example usage: test_shot_noise() without return types