code: def test_not_all_columns(self):
      test_df = pd.DataFrame(
          [[1, 2, 3], [4, np.NaN, 6]],
          index=[0, 1],
          columns=["a", "b", "c"],
      )
  
      self.assertRaises(
          ValueError,
          dataframe_functions.check_for_nans_in_columns,
          test_df,
      )
      self.assertRaises(
          ValueError,
          dataframe_functions.check_for_nans_in_columns,
          test_df,
          ["a", "b"],
      )
      self.assertRaises(
          ValueError,
          dataframe_functions.check_for_nans_in_columns,
          test_df,
          ["b"],
      )
      self.assertRaises(
          ValueError,
          dataframe_functions.check_for_nans_in_columns,
          test_df,
          "b",
      )
      self.assertRaises(
          ValueError,
          dataframe_functions.check_for_nans_in_columns,
          test_df,
          ["c", "b"],
      )
  
      dataframe_functions.check_for_nans_in_columns(
          test_df, columns=["a", "c"]
      )
      dataframe_functions.check_for_nans_in_columns(
          test_df, columns="a"
      )
  
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 CheckForNanTestCase obj's underlying member method code with example usage: obj.test_not_all_columns() without return types