code: def test_qm9():
      root = "./dataset/QM9"
  
      dataset = QM9(root, prop_name="penalized_logp")
  
      assert len(dataset) == 133885
      assert dataset.num_features == 4
      assert dataset.__repr__() == "qm9_property(133885)"
      assert len(dataset.get_split_idx()) == 2
  
      assert len(dataset[0]) == 6
      assert dataset[0].x.size() == (9, 4)
      assert dataset[0].y.size() == (1,)
      assert dataset[0].adj.size() == (4, 9, 9)
      assert dataset[0].bfs_perm_origin.size() == (9,)
      assert dataset[0].num_atom.size() == (1,)
  
      dataset = QM9(root, one_shot=True, prop_name="penalized_logp")
  
      assert len(dataset) == 133885
      assert dataset.__repr__() == "qm9_property(133885)"
      assert len(dataset.get_split_idx()) == 2
  
      assert len(dataset[0]) == 5
      assert dataset[0].x.size() == (5, 9)
      assert dataset[0].y.size() == (1,)
      assert dataset[0].adj.size() == (4, 9, 9)
      assert dataset[0].num_atom.size() == (1,)
  
      shutil.rmtree(root)
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_qm9() without return types