code: def test_restrict_dict(self):
      kind_to_df = {
          "a": pd.DataFrame({"id": [1, 2, 3]}),
          "b": pd.DataFrame({"id": [3, 4, 5]}),
      }
  
      kind_to_df_restricted = (
          dataframe_functions.restrict_input_to_index(
              kind_to_df, "id", [3]
          )
      )
      self.assertEqual(list(kind_to_df_restricted["a"].id), [3])
      self.assertEqual(list(kind_to_df_restricted["b"].id), [3])
  
      kind_to_df_restricted2 = (
          dataframe_functions.restrict_input_to_index(
              kind_to_df, "id", [1, 2, 3, 4, 5]
          )
      )
      self.assertTrue(
          kind_to_df_restricted2["a"].equals(kind_to_df["a"])
      )
      self.assertTrue(
          kind_to_df_restricted2["b"].equals(kind_to_df["b"])
      )
  
imports: import warnings
  from unittest import TestCase
  import numpy
  import pandas
  from pandas.testing import assert_frame_equal, assert_series_equal
  from tests.fixtures import warning_free
  from tsfresh import extract_relevant_features
  from tsfresh.feature_extraction.settings import MinimalFCParameters
  from tsfresh.utilities import dataframe_functions
label_desc: Use image node_id 2 for calling the RestrictTestCase obj's underlying member method code with example usage: obj.test_restrict_dict() without return types