Show end carbons test

cdk · Jun 11, 2011 · 10edd37 · 10edd37
1 parent 76bcd7e
commit 10edd37
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/src/test/org/openscience/cdk/renderer/generators/AbstractGeneratorTest.java b/src/test/org/openscience/cdk/renderer/generators/AbstractGeneratorTest.java
@@ -220,6 +220,16 @@ public IAtomContainer makeSingleBond() {
 		return container;
 	}
 
+	public IAtomContainer makeCCC() {
+	    IAtomContainer container = builder.newInstance(IAtomContainer.class);
+        container.addAtom(builder.newInstance(IAtom.class,"C", new Point2d(-1,-1)));
+        container.addAtom(builder.newInstance(IAtom.class,"C", new Point2d( 0, 0)));
+        container.addAtom(builder.newInstance(IAtom.class,"C", new Point2d( 1,-1)));
+        container.addBond(0, 1, IBond.Order.SINGLE);
+        container.addBond(1, 2, IBond.Order.SINGLE);
+        return container;
+	}
+
 	/**
 	 * Make a square (sort-of cyclobutane, without any hydrogens) centered on 
 	 * the origin, with a bond length of 2.

diff --git a/src/test/org/openscience/cdk/renderer/generators/BasicAtomGeneratorTest.java b/src/test/org/openscience/cdk/renderer/generators/BasicAtomGeneratorTest.java
@@ -181,6 +181,15 @@ public void kekuleTest() {
         Assert.assertEquals(0, getAllSimpleElements(generator, singleBond).size());
 	}
 
+	@Test
+    public void showEndCarbonsTest() {
+        IAtomContainer singleBond = makeCCC();
+        model.set(ShowEndCarbons.class, true);
+        Assert.assertEquals(2, getAllSimpleElements(generator, singleBond).size());
+        model.set(ShowEndCarbons.class, false);
+        Assert.assertEquals(0, getAllSimpleElements(generator, singleBond).size());
+    }
+
 	@Test
 	public void testSingleAtom() {
 		IAtomContainer singleAtom = makeSingleAtom();