code: def test_get_draws(self, data):
      fit = data.obj
      if pystan_version() == 2:
          draws, _ = get_draws(fit, variables=["theta", "theta"])
      else:
          draws, _ = get_draws_stan3(fit, variables=["theta", "theta"])
      assert draws.get("theta") is not None
  
imports: import importlib
  from collections import OrderedDict
  import numpy
  import pytest
  from ...None import from_pystan
  from ...data.io_pystan import get_draws, get_draws_stan3
  from ..helpers import chains, check_multiple_attrs, draws, eight_schools_params, importorskip, load_cached_models, pystan_version, running_on_ci
label_desc: Use image node_id 11 for calling the TestDataPyStan obj's underlying member method code with example usage: obj.test_get_draws(data) without return types