code: def test_just_dance_comment_analysis():
      # Arrange
      train, test = just_dance_comment_analysis.load_data(
          data_format="Dataframe", as_train_test=True
      )
      full = just_dance_comment_analysis.load_data(
          data_format="Dataframe", as_train_test=False
      )
      full_ds = just_dance_comment_analysis.load_data(
          data_format="TextData",
          as_train_test=False,
          include_embeddings=True,
      )
      preds = (
          just_dance_comment_analysis.load_precalculated_predictions(
              pred_format="predictions", as_train_test=False
          )
      )
      probas = (
          just_dance_comment_analysis.load_precalculated_predictions(
              pred_format="probabilities", as_train_test=False
          )
      )
      properties = just_dance_comment_analysis.load_properties(
          as_train_test=False
      )
      (
          train_props,
          test_props,
      ) = just_dance_comment_analysis.load_properties(
          as_train_test=True
      )
      embeddings = just_dance_comment_analysis.load_embeddings(
          as_train_test=False
      )
      (
          train_embeddings,
          test_embeddings,
      ) = just_dance_comment_analysis.load_embeddings(
          as_train_test=True
      )
  
      # Act & Assert
      assert_that(len(train) + len(test), equal_to(len(full)))
      assert_that(train.columns, contains_exactly(*test.columns))
      assert_that(train.columns, contains_exactly(*full.columns))
  
      assert_that(len(full_ds.text), equal_to(len(full)))
      assert_that(len(full_ds.text), equal_to(len(preds)))
      assert_that(len(full_ds.text), equal_to(len(probas)))
  
      assert_that(len(properties), equal_to(len(full)))
      assert_that(
          len(train_props) + len(test_props), equal_to(len(full))
      )
      assert_that(len(train_props), equal_to(len(train)))
  
      assert_that(len(embeddings), equal_to(len(full)))
      assert_that(
          len(train_embeddings) + len(test_embeddings),
          equal_to(len(full)),
      )
      assert_that(len(train_embeddings), equal_to(len(train)))
      assert_that(embeddings.shape, contains_exactly(16281, 1536))
      assert_that(train_embeddings.shape, contains_exactly(7669, 1536))
      assert_that(test_embeddings.shape, contains_exactly(8612, 1536))
  
imports: import numpy
  from hamcrest import assert_that, contains_exactly, equal_to
  from deepchecks.nlp.datasets.classification import just_dance_comment_analysis, tweet_emotion
  from deepchecks.nlp.datasets.token_classification import scierc_ner
label_desc: Use image node_id 2 for calling a global function with example usage: test_just_dance_comment_analysis() without return types