code: def test_fit_array_draw_contents(fit):
      """
      Make sure shapes are getting unraveled correctly. Mixing up row-major and
      column-major data is a potential issue.
      """
      beta = fit["beta"]
      assert beta.shape == (K, 1, 2, num_samples * num_chains)
      beta_mean = np.mean(beta, axis=-1)
      assert beta_mean.shape == (K, 1, 2)
      assert np.all(beta_mean[:, 0, 0] < 4)
      assert np.all(beta_mean[:, 0, 1] > 99)
  
imports: import numpy
  import pytest
  import stan
label_desc: Use image node_id 3 for calling a global function with example usage: test_fit_array_draw_contents(fit) without return types