code: def test_get_diffusion_beta_schedule(schedule_type):
      out = get_diffusion_beta_schedule(
          schedule_type,
          beta_start=0.1,
          beta_end=0.2,
          num_diffusion_timesteps=10,
      )
      assert out.size() == (10,)
  
imports: import pytest
  import torch
  from torch_geometric.utils.noise_scheduler import get_diffusion_beta_schedule, get_smld_sigma_schedule
label_desc: Use image node_id 2 for calling a global function with example usage: test_get_diffusion_beta_schedule(schedule_type) without return types