code: def test_set_compute_func_at(
      self, shared_step, dataplaceholders, teardown
  ):
      shared_step.set_compute_func_at(1, shared_step.predict)
      assert shared_step.get_compute_func_at(1) == shared_step.predict
  
imports: from contextlib import contextmanager
  from functools import partial
  import pytest
  from sklearn.base import TransformerMixin, BaseEstimator
  from baikal import Input, Step, set_config
  from baikal._core.data_placeholder import DataPlaceholder
  from baikal._core.step import InputStep
  from tests.helpers.fixtures import teardown
  from tests.helpers.dummy_steps import DummyMIMO, DummySISO, DummyEstimator
  from tests.helpers.sklearn_steps import LogisticRegression, PCA
label_desc: Use image node_id 29 for calling the TestStep obj's underlying member method code with example usage: obj.test_set_compute_func_at(shared_step, dataplaceholders, teardown) without return types