JAL-3551 the correct PyMOL colour by chain command

[jalview.git] / test / jalview / ext / pymol / PymolCommandsTest.java

diff --git a/test/jalview/ext/pymol/PymolCommandsTest.java b/test/jalview/ext/pymol/PymolCommandsTest.java
index a66f809..9dc107e 100644 (file)
--- a/test/jalview/ext/pymol/PymolCommandsTest.java
+++ b/test/jalview/ext/pymol/PymolCommandsTest.java
@@ -186,6 +186,18 @@ public class PymolCommandsTest
   }
 
   @Test(groups = "Functional")
+  public void testColourByCharge()
+  {
+    PymolCommands testee = new PymolCommands();
+    List<StructureCommandI> cmds = testee.colourByCharge();
+    assertEquals(cmds.size(), 4);
+    assertEquals(cmds.get(0).toString(), "color(white,*)");
+    assertEquals(cmds.get(1).toString(), "color(red,resn ASP resn GLU)");
+    assertEquals(cmds.get(2).toString(), "color(blue,resn LYS resn ARG)");
+    assertEquals(cmds.get(3).toString(), "color(yellow,resn CYS)");
+  }
+
+  @Test(groups = "Functional")
   public void testOpenCommandFile()
   {
     PymolCommands testee = new PymolCommands();
@@ -202,6 +214,13 @@ public class PymolCommandsTest
   }
 
   @Test(groups = "Functional")
+  public void testColourByChain()
+  {
+    PymolCommands testee = new PymolCommands();
+    assertEquals(testee.colourByChain().toString(), "spectrum(chain)");
+  }
+
+  @Test(groups = "Functional")
   public void testGetColourCommand()
   {
     PymolCommands testee = new PymolCommands();