code: def test_expected_output(dim, percents, in_tensor, out_tensors):
      """Test with a few expected input & outputs."""
      ecp = ECP(dim, percents)
      out = ecp.measure(in_tensor)
      objects_are_equal(
          out, out_tensors, raise_exception=True, rtol=RTOL, atol=ATOL
      )
      out_fft = ecp.measure_fft(torch.fft.ifft(in_tensor, dim=dim))
      objects_are_equal(
          out_fft,
          out_tensors,
          raise_exception=True,
          rtol=RTOL,
          atol=ATOL,
      )
  
file_path: fairscale/tests/experimental/wgit/test_signal_sparsity_profiling.py
imports: import time
  import pytest
  import torch
  from fairscale.experimental.wgit.signal_sparsity_profiling import EnergyConcentrationProfile
  from fairscale.fair_dev.testing.testing import objects_are_equal, skip_if_no_cuda
label_desc: Use image node_id 4 for calling a global function with example usage: test_expected_output(dim, percents, in_tensor, out_tensors) without return types