code: def test_load_data():
      train_data = mz.datasets.wiki_qa.load_data(
          "train", task="ranking"
      )
      assert len(train_data) == 20360
      train_data, _ = mz.datasets.wiki_qa.load_data(
          "train", task="classification", return_classes=True
      )
      assert len(train_data) == 20360
  
      dev_data = mz.datasets.wiki_qa.load_data(
          "dev", task="ranking", filtered=False
      )
      assert len(dev_data) == 2733
      dev_data, tag = mz.datasets.wiki_qa.load_data(
          "dev",
          task="classification",
          filtered=True,
          return_classes=True,
      )
      assert len(dev_data) == 1126
      assert tag == [False, True]
  
      test_data = mz.datasets.wiki_qa.load_data(
          "test", task="ranking", filtered=False
      )
      assert len(test_data) == 6165
      test_data, tag = mz.datasets.wiki_qa.load_data(
          "test",
          task="classification",
          filtered=True,
          return_classes=True,
      )
      assert len(test_data) == 2341
      assert tag == [False, True]
  
imports: import pytest
  import matchzoo
label_desc: Use image node_id 1 for calling a global function with example usage: test_load_data() without return types