code: def test2_poisson_predict_length():
      """
      Tests that the length of the predict dataframe is equal to no of steps h
      """
      model = pf.GASX(
          formula="y ~ x1 + x2",
          data=data,
          ar=1,
          sc=1,
          family=pf.Poisson(),
      )
      x = model.fit()
      x.summary()
      assert model.predict(h=5, oos_data=data_oos).shape[0] == 5
  
imports: import numpy
  import pandas
  import pyflux
label_desc: Use image node_id 33 for calling a global function with example usage: test2_poisson_predict_length() without return types