code: def test_bbvi_mini_batch():
      """
      Tests an ARIMA model estimated with BBVI and that the length of the latent variable
      list is correct, and that the estimated latent variables are not nan
      """
      model = ARIMA(data=data, ar=1, ma=1)
      x = model.fit(
          "BBVI", iterations=100, quiet_progress=True, mini_batch=32
      )
      assert len(model.latent_variables.z_list) == 4
      lvs = np.array([i.value for i in model.latent_variables.z_list])
      assert len(lvs[np.isnan(lvs)]) == 0
  
file_path: pyflux/pyflux/arma/tests/test_arima_inference.py
imports: import numpy
  from pyflux.arma import ARIMA
label_desc: Use image node_id 2 for calling a global function with example usage: test_bbvi_mini_batch() without return types