code: def test_time_domain_criterion_forward(criterion_class):
      criterion = criterion_class()
  
      batch = 2
      inf = torch.rand(batch, 2000)
      ref = torch.rand(batch, 2000)
  
      loss = criterion(ref, inf)
      assert loss.shape == (batch,), (
          "Invlid loss shape with " + criterion.name
      )
  
imports: import pytest
  import torch
  from packaging.version import parse
  from espnet2.enh.loss.criterions.time_domain import CISDRLoss, MultiResL1SpecLoss, SDRLoss, SISNRLoss, SNRLoss, TimeDomainL1, TimeDomainMSE
label_desc: Use image node_id 1 for calling a global function with example usage: test_time_domain_criterion_forward(criterion_class) without return types