code: def test_context_window(device):
      from speechbrain.processing.features import ContextWindow
  
      inp = (
          torch.tensor([1, 2, 3], device=device)
          .unsqueeze(0)
          .unsqueeze(-1)
          .float()
      )
      compute_cw = ContextWindow(left_frames=1, right_frames=1).to(
          device
      )
      out = (
          torch.tensor([[0, 1, 2], [1, 2, 3], [2, 3, 0]], device=device)
          .unsqueeze(0)
          .float()
      )
      assert torch.sum(compute_cw(inp) == out) == 9
  
      inp = torch.rand([2, 10, 5], device=device)
      compute_cw = ContextWindow(left_frames=0, right_frames=0).to(
          device
      )
      assert torch.sum(compute_cw(inp) == inp) == inp.numel()
  
      assert torch.jit.trace(compute_cw, inp)
  
imports: import torch
label_desc: Use image node_id 2 for calling a global function with example usage: test_context_window(device) without return types