code: def test_2_predict_nans():
      """
      Tests that the predictions are not NaNs
      """
      model = ARIMAX(formula="y ~ x1 + x2", data=data, ar=2, ma=2)
      x = model.fit()
      x.summary()
      assert (
          len(
              model.predict(h=5, oos_data=data_oos).values[
                  np.isnan(model.predict(h=5, oos_data=data_oos).values)
              ]
          )
          == 0
      )
  
imports: import numpy
  import pandas
  from pyflux.arma import ARIMAX
label_desc: Use image node_id 22 for calling a global function with example usage: test_2_predict_nans() without return types