code: def test_fit_matrix_draw_order(fit):
      assert fit is not None
      offset = len(fit.sample_and_sampler_param_names)
      assert fit._draws.shape == (
          offset + K * D,
          num_samples,
          num_chains,
      )
      assert len(fit._draws[:, 0, 0]) == offset + K * D
      assert fit._parameter_indexes("beta") == tuple(
          offset + i for i in range(K * D)
      )
  
imports: import pytest
  import stan
label_desc: Use image node_id 3 for calling a global function with example usage: test_fit_matrix_draw_order(fit) without return types