code: def test_base_n_features_in(datatype, use_integer_n_features):
      X_train, _, _, _ = small_classification_dataset(datatype)
      integer_n_features = 8
      clf = cuml.Base()
  
      if use_integer_n_features:
          clf._set_n_features_in(integer_n_features)
          assert clf.n_features_in_ == integer_n_features
      else:
          clf._set_n_features_in(X_train)
          assert clf.n_features_in_ == X_train.shape[1]
  
file_path: cuml/python/cuml/tests/test_base.py
imports: import inspect
  import cuml
  import pytest
  import numpydoc.docscrape
  from pylibraft.common.cuda import Stream
  from cuml.testing.utils import get_classes_from_package, small_classification_dataset
  from cuml._thirdparty.sklearn.utils.skl_dependencies import BaseEstimator
label_desc: Use image node_id 4 for calling a global function with example usage: test_base_n_features_in(datatype, use_integer_n_features) without return types