code: def test_mask_label():
      model = MaskLabel(2, 10)
      assert str(model) == "MaskLabel()"
  
      x = torch.rand(4, 10)
      y = torch.tensor([1, 0, 1, 0])
      mask = torch.tensor([False, False, True, True])
  
      out = model(x, y, mask)
      assert out.size() == (4, 10)
      assert torch.allclose(out[~mask], x[~mask])
  
      model = MaskLabel(2, 10, method="concat")
      out = model(x, y, mask)
      assert out.size() == (4, 20)
      assert torch.allclose(out[:, :10], x)
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_mask_label() without return types