@Test
public void testGetAtomColor() {

    CDK2DAtomColors colors = new CDK2DAtomColors();
    Assert.assertNotNull(colors);
    IAtom hydrogen = new Atom("H");
    hydrogen.setAtomicNumber(1);
    Assert.assertEquals(new Color(0, 0, 0), colors.getAtomColor(hydrogen));
    IAtom helium = new Atom("He");
    helium.setAtomicNumber(2);
    Assert.assertEquals(new Color(217, 255, 255), colors.getAtomColor(helium));
}