def test_collect_linker_atoms():
    mol = Chem.MolFromSmiles('CCCCCCCCCc1ccccc1')
    remove_atoms = set()
    a = collect_linker_atoms(mol.GetAtomWithIdx(0), remove_atoms, True)
    assert (len(a) == 1)
    assert (len(remove_atoms) == 9)
    remove_atoms.clear()
    a = collect_linker_atoms(mol.GetAtomWithIdx(0), remove_atoms, False)
    assert (len(a) == 1)
    assert (len(remove_atoms) == 8)
def test_collect_linker_atoms():
    mol = Chem.MolFromSmiles('CCCCCCCCCc1ccccc1')
    remove_atoms = set()
    a = collect_linker_atoms(mol.GetAtomWithIdx(0), remove_atoms, True)
    assert (len(a) == 1)
    assert (len(remove_atoms) == 9)
    remove_atoms.clear()
    a = collect_linker_atoms(mol.GetAtomWithIdx(0), remove_atoms, False)
    assert (len(a) == 1)
    assert (len(remove_atoms) == 8)