code: def test_adaptation_schedule(
      adapt_step_size, adapt_mass, warmup_steps, expected
  ):
      adapter = WarmupAdapter(
          0.1,
          adapt_step_size=adapt_step_size,
          adapt_mass_matrix=adapt_mass,
      )
      adapter.configure(warmup_steps, mass_matrix_shape={"z": (5, 5)})
      expected_schedule = [adapt_window(i, j) for i, j in expected]
      assert_equal(
          adapter.adaptation_schedule, expected_schedule, prec=0
      )
  
imports: import pytest
  import torch
  from pyro.infer.mcmc.adaptation import ArrowheadMassMatrix, BlockMassMatrix, WarmupAdapter, adapt_window
  from tests.common import assert_close, assert_equal
label_desc: Use image node_id 1 for calling a global function with example usage: test_adaptation_schedule(adapt_step_size, adapt_mass, warmup_steps, expected) without return types