code: def test_variance(self):
      scheduler_class = self.scheduler_classes[0]
      scheduler_config = self.get_scheduler_config()
      scheduler = scheduler_class(**scheduler_config)
      state = scheduler.create_state()
  
      assert (
          jnp.sum(jnp.abs(scheduler._get_variance(state, 0, 0) - 0.0))
          < 1e-5
      )
      assert (
          jnp.sum(
              jnp.abs(
                  scheduler._get_variance(state, 420, 400) - 0.14771
              )
          )
          < 1e-5
      )
      assert (
          jnp.sum(
              jnp.abs(
                  scheduler._get_variance(state, 980, 960) - 0.32460
              )
          )
          < 1e-5
      )
      assert (
          jnp.sum(jnp.abs(scheduler._get_variance(state, 0, 0) - 0.0))
          < 1e-5
      )
      assert (
          jnp.sum(
              jnp.abs(
                  scheduler._get_variance(state, 487, 486) - 0.00979
              )
          )
          < 1e-5
      )
      assert (
          jnp.sum(
              jnp.abs(scheduler._get_variance(state, 999, 998) - 0.02)
          )
          < 1e-5
      )
  
imports: import inspect
  import tempfile
  import unittest
  from typing import Dict, List, Tuple
  from diffusers import FlaxDDIMScheduler, FlaxDDPMScheduler, FlaxPNDMScheduler
  from diffusers.utils import is_flax_available
  from diffusers.utils.testing_utils import require_flax
label_desc: Use image node_id 14 for calling the FlaxDDIMSchedulerTest obj's underlying member method code with example usage: obj.test_variance() without return types