code: def test_log_shape(shape, capsys, test_df):
      """Test logging of shape can be switched on and off"""
  
      @log_step(shape=shape)
      def do_nothing(df, *args, **kwargs):
          return df
  
      test_df.pipe(do_nothing)
  
      captured = capsys.readouterr()
  
      assert (f"n_obs={test_df.shape[0]}" in captured.out) == shape
      assert (f"n_col={test_df.shape[1]}" in captured.out) == shape
  
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 12 for calling a global function with example usage: test_log_shape(shape, capsys, test_df) and returns: df