@Test
public void testGetAtomColor() {

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