chemsequencemanipulatortest.java

化学图形处理软件

/* $Revision: 7691 $ $Author: egonw $ $Date: 2007-01-11 12:47:48 +0100 (Thu, 11 Jan 2007) $
 * 
 * Copyright (C) 2007  Egon Willighagen <[e-mail]>
 * 
 * Contact: [e-mail]
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public License
 * as published by the Free Software Foundation; either version 2.1
 * of the License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 
 */
package org.openscience.cdk.test.tools.manipulator;

import java.util.Iterator;
import java.util.List;

import junit.framework.Test;
import junit.framework.TestSuite;

import org.openscience.cdk.Atom;
import org.openscience.cdk.Bond;
import org.openscience.cdk.ChemModel;
import org.openscience.cdk.ChemSequence;
import org.openscience.cdk.Molecule;
import org.openscience.cdk.MoleculeSet;
import org.openscience.cdk.Reaction;
import org.openscience.cdk.ReactionSet;
import org.openscience.cdk.interfaces.IAtom;
import org.openscience.cdk.interfaces.IBond;
import org.openscience.cdk.interfaces.IChemModel;
import org.openscience.cdk.interfaces.IChemSequence;
import org.openscience.cdk.interfaces.IMolecule;
import org.openscience.cdk.interfaces.IMoleculeSet;
import org.openscience.cdk.interfaces.IReaction;
import org.openscience.cdk.interfaces.IReactionSet;
import org.openscience.cdk.test.CDKTestCase;
import org.openscience.cdk.tools.manipulator.ChemSequenceManipulator;

/**
 * @cdk.module test-standard
 */
public class ChemSequenceManipulatorTest extends CDKTestCase {
    
	IMolecule molecule1 = null;
	IMolecule molecule2 = null;
	IAtom atomInMol1 = null;
	IBond bondInMol1 = null;
	IAtom atomInMol2 = null;
	IMoleculeSet moleculeSet = null;
	IReaction reaction = null;
	IReactionSet reactionSet = null;
	IChemModel chemModel1 = null;
	IChemModel chemModel2 = null;
	IChemSequence chemSequence = null;

	
    public ChemSequenceManipulatorTest(String name) {
        super(name);
    }
    
    public void setUp() {
		molecule1 = new Molecule();
		atomInMol1 = new Atom("Cl");
		molecule1.addAtom(atomInMol1);
		molecule1.addAtom(new Atom("Cl"));
		bondInMol1 = new Bond(atomInMol1, molecule1.getAtom(1));
		molecule1.addBond(bondInMol1);
		molecule2 = new Molecule();
		atomInMol2 = new Atom("O");
		atomInMol2.setHydrogenCount(2);
		molecule2.addAtom(atomInMol2);
		moleculeSet = new MoleculeSet();
		moleculeSet.addAtomContainer(molecule1);
		moleculeSet.addAtomContainer(molecule2);
		reaction = new Reaction();
		reaction.addReactant(molecule1);
		reaction.addProduct(molecule2);
		reactionSet = new ReactionSet();
		reactionSet.addReaction(reaction);
		chemModel1 = new ChemModel();
		chemModel1.setMoleculeSet(moleculeSet);
		chemModel2 = new ChemModel();
		chemModel2.setReactionSet(reactionSet);
		chemSequence = new ChemSequence();
		chemSequence.addChemModel(chemModel1);
		chemSequence.addChemModel(chemModel2);
	}
    
	public static Test suite() {
		return new TestSuite(ChemSequenceManipulatorTest.class);
	}

	public void testGetAtomCount_IChemSequence() {
		int count = ChemSequenceManipulator.getAtomCount(chemSequence);
    	assertEquals(6, count);
	}
	
	public void testGetBondCount_IChemSequence()
	{
		int count = ChemSequenceManipulator.getBondCount(chemSequence);
    	assertEquals(2, count);
	}
	
	public void testGetAllAtomContainers_IChemSequence()
	{
		List list = ChemSequenceManipulator.getAllAtomContainers(chemSequence);
		assertEquals(4, list.size());
	}
	
	public void testGetAllChemObjects_IChemSequence()
	{
		List list = ChemSequenceManipulator.getAllChemObjects(chemSequence);
    	int molCount = 0;
    	int molSetCount = 0;
    	int reactionCount = 0;
    	int reactionSetCount = 0;
    	int chemModelCount = 0;
    	for (Iterator iter = list.iterator(); iter.hasNext();) {
    		Object o = iter.next();
    		//if (o instanceof IAtom) ++atomCount;
    		//if (o instanceof IBond) ++bondCount;
    		if (o instanceof IMolecule) ++molCount;
    		else if (o instanceof IMoleculeSet) ++molSetCount;
    		else if (o instanceof IReaction) ++reactionCount;
    		else if (o instanceof IReactionSet) ++reactionSetCount;
    		else if (o instanceof IChemModel) ++chemModelCount;
    		else fail("Unexpected Object of type " + o.getClass());
    	}
    	//assertEquals(3, atomCount);
    	//assertEquals(1, bondCount);
    	assertEquals(2, molCount);
    	assertEquals(1, molSetCount);
    	assertEquals(1, reactionCount);
    	assertEquals(1, reactionSetCount);
    	assertEquals(2, chemModelCount);
	}

	
	
}