code: def test_ba_shapes(get_dataset):
      with pytest.warns(UserWarning, match="is deprecated"):
          dataset = get_dataset(name="BAShapes")
  
      assert str(dataset) == "BAShapes()"
      assert len(dataset) == 1
      assert dataset.num_features == 10
      assert dataset.num_classes == 4
  
      data = dataset[0]
      assert len(data) == 5
      assert data.edge_index.size(1) >= 1120
      assert data.x.size() == (700, 10)
      assert data.y.size() == (700,)
      assert data.expl_mask.sum() == 60
      assert data.edge_label.sum() == 960
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_ba_shapes(get_dataset) without return types