code: def test_set_lags_when_lags_argument_is_int():
      """
      Test how lags and max_lag attributes change when lags argument is integer
      positive (5).
      """
      forecaster = ForecasterAutoregDirect(
          LinearRegression(), lags=3, steps=2
      )
      forecaster.set_lags(lags=5)
  
      assert (forecaster.lags == np.array([1, 2, 3, 4, 5])).all()
      assert forecaster.max_lag == 5
  
file_path: skforecast/skforecast/ForecasterAutoregDirect/tests/test_set_lags.py
imports: import re
  import pytest
  import numpy
  import pandas
  from skforecast.ForecasterAutoregDirect import ForecasterAutoregDirect
  from sklearn.linear_model import LinearRegression
label_desc: Use image node_id 1 for calling a global function with example usage: test_set_lags_when_lags_argument_is_int() without return types