code: def test_get_features_in_step(
      data: np.ndarray, expected: Union[float, dict, bool]
  ):
      out = get_features_in_step(
          max_steps_per_input=data["max_steps_per_input"],
          input_shape=data["input_shape"],
      )
      assert out == expected, "Test failed."
  
imports: import pytest
  from pytest_lazyfixture import lazy_fixture
  from quantus.helpers.utils import *
label_desc: Use image node_id 27 for calling a global function with example usage: test_get_features_in_step(data, expected) without return types