code: def test_predict_length():
      """
      Tests that the prediction dataframe length is equal to the number of steps h
      """
      model = ARIMA(data=data, ar=2, ma=2, family=Skewt())
      x = model.fit()
      assert model.predict(h=5).shape[0] == 5
  
file_path: pyflux/pyflux/arma/tests/test_arima_skewt.py
imports: import numpy
  from pyflux.arma import ARIMA
  from pyflux.families import Skewt
label_desc: Use image node_id 4 for calling a global function with example usage: test_predict_length() without return types