@Test
public void getAtomColorTest() {

    Color testColor = Color.RED;
    IAtomContainer singleAtom = makeSingleAtom("O");
    model.set(AtomColor.class, testColor);
    model.set(ColorByType.class, false);
    generator.getAtomColor(singleAtom.getAtom(0), model);
    List<IRenderingElement> elements = getAllSimpleElements(generator, singleAtom);
    Assert.assertEquals(1, elements.size());
    AtomSymbolElement element = ((AtomSymbolElement) unbox(elements.get(0)));
    Assert.assertEquals(testColor, element.color);
}