code: def test_dblock():
      inp = torch.rand(32, 50, 130)
      layer = DBlock(50, 100, 2)
      o = layer(inp)
  
      assert o.shape[0] == 32
      assert o.shape[1] == 100
      assert o.shape[2] == 65
      assert isinstance(o, torch.FloatTensor)
  
      layer.apply_weight_norm()
      layer.remove_weight_norm()
  
file_path: TTS/tests/vocoder_tests/test_wavegrad_layers.py
imports: import torch
  from TTS.vocoder.configs import WavegradConfig
  from TTS.vocoder.layers.wavegrad import DBlock, FiLM, PositionalEncoding, UBlock
  from TTS.vocoder.models.wavegrad import Wavegrad, WavegradArgs
label_desc: Use image node_id 4 for calling a global function with example usage: test_dblock() without return types