AtomSignature extends AbstractVertexSignature { @Override protected int convertEdgeLabelToColor(String edgeLabel) { if (edgeLabel.equals("")) { return 1; } else if (edgeLabel.equals("=")) { return 2; } else if (edgeLabel.equals("#")) { return 3; } else if (edgeLabel.equals("$")) { return 4; } else if (edgeLabel.equals("p")) { return 5; } return 0; } AtomSignature(int atomIndex, IAtomContainer molecule);  AtomSignature(IAtom atom, IAtomContainer molecule);  AtomSignature(int atomIndex, int height, IAtomContainer molecule);  AtomSignature(IAtom atom, int height, IAtomContainer molecule);  AtomSignature(int atomIndex, int height, InvariantType invariantType, IAtomContainer molecule);  AtomSignature(IAtom atom, int height, InvariantType invariantType, IAtomContainer molecule);   }
@Test public void convertEdgeLabelToColorTest() { IAtomContainer ac = makeBenzene(); AtomSignature atomSignature = new AtomSignature(0, ac); int aromaticColor = atomSignature.convertEdgeLabelToColor("p"); Assert.assertTrue(aromaticColor > 0); int singleColor = atomSignature.convertEdgeLabelToColor(""); int doubleColor = atomSignature.convertEdgeLabelToColor("="); int tripleColor = atomSignature.convertEdgeLabelToColor("#"); Assert.assertTrue(singleColor < doubleColor); Assert.assertTrue(doubleColor < tripleColor); }