code: def test_set_embeddings(text_classification_dataset_mock):
      # Arrange
      dataset = text_classification_dataset_mock
      embeddings = pd.DataFrame({"0": [1, 2, 3], "1": [4, 5, 6]})
      assert_that(
          dataset._embeddings, equal_to(None)
      )  # pylint: disable=protected-access
  
      dataset.set_embeddings(embeddings)
      assert_that(
          (dataset.embeddings != embeddings).sum().sum(), equal_to(0)
      )
  
      # Check that works for np.array:
      dataset._embeddings = None  # pylint: disable=protected-access
  
      embeddings = np.array([[1, 2, 3], [4, 5, 6]]).T
  
      dataset.set_embeddings(embeddings)
      assert_that(
          (dataset.embeddings != embeddings).sum().sum(), equal_to(0)
      )
  
imports: import numpy
  import pandas
  from hamcrest import assert_that, calling, contains_exactly, equal_to, raises
  from deepchecks.core.errors import DeepchecksValueError
  from deepchecks.nlp.text_data import TextData
label_desc: Use image node_id 11 for calling a global function with example usage: test_set_embeddings(text_classification_dataset_mock) without return types