code: def test_numba_stats():
      """Numba test for r2_score"""
      state = Numba.numba_flag  # Store the current state of Numba
      set_1 = np.random.randn(100, 100)
      set_2 = np.random.randn(100, 100)
      set_3 = np.random.rand(100)
      set_4 = np.random.rand(100)
      Numba.disable_numba()
      non_numba = r2_score(set_1, set_2)
      non_numba_one_dimensional = r2_score(set_3, set_4)
      Numba.enable_numba()
      with_numba = r2_score(set_1, set_2)
      with_numba_one_dimensional = r2_score(set_3, set_4)
      assert (
          state == Numba.numba_flag
      )  # Ensure that initial state = final state
      assert np.allclose(non_numba, with_numba)
      assert np.allclose(
          non_numba_one_dimensional, with_numba_one_dimensional
      )
  
file_path: arviz/arviz/tests/base_tests/test_stats_numba.py
imports: import importlib
  import numpy
  import pytest
  from ...rcparams import rcParams
  from ...stats import r2_score, summary
  from ...utils import Numba
  from ..helpers import check_multiple_attrs, multidim_models, running_on_ci
  from .test_stats import centered_eight, non_centered_eight
label_desc: Use image node_id 2 for calling a global function with example usage: test_numba_stats() without return types