code: def test_multiple_vars(self):
      A = np.arange(1000)
      B = np.arange(1000)
      mA, mB = pm.Minibatch(A, B, batch_size=10)
  
      [draw_mA, draw_mB] = pm.draw([mA, mB])
      assert draw_mA.shape == (10,)
      np.testing.assert_allclose(draw_mA, draw_mB)
  
      # Check invalid dims
      A = np.arange(1000)
      C = np.arange(999)
      mA, mC = pm.Minibatch(A, C, batch_size=10)
  
      with pytest.raises(
          AssertionError,
          match=re.escape(
              "All variables shape[0] in Minibatch should be equal"
          ),
      ):
          pm.draw([mA, mC])
  
imports: import io
  import itertools
  import re
  import cloudpickle
  import numpy
  import pytensor
  import pytensor.tensor
  import pytest
  from pytensor import shared
  from pytensor.tensor.variable import TensorVariable
  import pymc
  from pymc.data import is_minibatch
  from pymc.pytensorf import GeneratorOp, floatX
label_desc: Use image node_id 6 for calling the TestMinibatch obj's underlying member method code with example usage: obj.test_multiple_vars() without return types