JAL-1827 code and comment tidy up

[jalview.git] / test / jalview / ext / jmol / PDBFileWithJmolTest.java

diff --git a/test/jalview/ext/jmol/PDBFileWithJmolTest.java b/test/jalview/ext/jmol/PDBFileWithJmolTest.java
index fd02d00..4dc7095 100644 (file)
--- a/test/jalview/ext/jmol/PDBFileWithJmolTest.java
+++ b/test/jalview/ext/jmol/PDBFileWithJmolTest.java
@@ -33,6 +33,7 @@ import jalview.io.FileLoader;
 
 import java.util.Vector;
 
+import org.jmol.c.STR;
 import org.testng.annotations.BeforeMethod;
 import org.testng.annotations.Test;
 
@@ -206,4 +207,35 @@ public class PDBFileWithJmolTest
     assertEquals("ALC", seqs.get(0).getSequenceAsString());
     assertEquals("ALC", mcseqs.get(0).getSequenceAsString());
   }
+
+  @Test(groups = "Functional")
+  public void testSetSecondaryStructure()
+  {
+    PDBFileWithJmol testee = new PDBFileWithJmol();
+    char[] struct = new char[10];
+    char[] structCode = new char[10];
+    struct[0] = '1';
+    structCode[0] = '1';
+
+    testee.setSecondaryStructure(STR.NONE, 0, struct, structCode);
+    testee.setSecondaryStructure(STR.HELIX, 1, struct, structCode);
+    testee.setSecondaryStructure(STR.HELIX310, 2, struct, structCode);
+    testee.setSecondaryStructure(STR.HELIXALPHA, 3, struct, structCode);
+    testee.setSecondaryStructure(STR.HELIXPI, 4, struct, structCode);
+    testee.setSecondaryStructure(STR.SHEET, 5, struct, structCode);
+
+    assertEquals(0, struct[0]);
+    assertEquals('H', struct[1]);
+    assertEquals('3', struct[2]);
+    assertEquals('H', struct[3]);
+    assertEquals('P', struct[4]);
+    assertEquals('E', struct[5]);
+
+    assertEquals(0, structCode[0]);
+    assertEquals('H', structCode[1]);
+    assertEquals('H', structCode[2]);
+    assertEquals('H', structCode[3]);
+    assertEquals('H', structCode[4]);
+    assertEquals('E', structCode[5]);
+  }
 }