code: def test_bbvi_mini_batch_elbo():
      """
      Tests that the ELBO increases
      """
      model = ARIMAX(formula="y ~ x1", data=data, ar=1, ma=1)
      x = model.fit(
          "BBVI",
          iterations=200,
          mini_batch=32,
          record_elbo=True,
          quiet_progress=True,
      )
      assert x.elbo_records[-1] > x.elbo_records[0]
  
file_path: pyflux/pyflux/arma/tests/test_arimax_inference.py
imports: import numpy
  import pandas
  from pyflux.arma import ARIMAX
label_desc: Use image node_id 4 for calling a global function with example usage: test_bbvi_mini_batch_elbo() without return types