code: def test_add_lags_correct_X(test_X):
      expected = np.array([[-4, 2, -2, 3, 0, -6]])
      assert (add_lags(test_X, [0, 1], [1, 2]) == expected).all()
  
imports: import logging
  import numpy
  import pandas
  import pytest
  from sklego.pandas_utils import _add_lagged_numpy_columns, _add_lagged_pandas_columns, add_lags, log_step, log_step_extra
label_desc: Use image node_id 5 for calling a global function with example usage: test_add_lags_correct_X(test_X) without return types