def METHOD_NAME():
    """Test on carbon nitrogen molecule"""
    # Note there is a central nitrogen of degree 4, with 4 carbons
    # of degree 1 (connected only to central nitrogen).
    mols = ['C[N+](C)(C)C']
    # import rdkit.Chem
    # mols = [rdkit.Chem.MolFromSmiles(s) for s in raw_smiles]
    featurizer = dc.feat.WeaveFeaturizer()
    mols = featurizer.featurize(mols)
    mol = mols[0]
    # 5 atoms in compound
    assert mol.get_num_atoms() == 5
    # Test feature sizes
    assert mol.get_num_features() == 75
    # Should be a 3x3 interaction grid
    assert mol.get_pair_features().shape == (5 * 5, 14)