X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=test%2Fjalview%2Fschemes%2FPIDColourSchemeTest.java;h=c2f86d6e24b1602065e8770fdb9e73dac3e0e522;hb=1deb45d06fc8fa719bd19b25959397d8b3fd9274;hp=73b270f8353474918cd32b846a280dc76b3940e0;hpb=4f094851cd2247ea2d982e7829a6f5b663042573;p=jalview.git

diff --git a/test/jalview/schemes/PIDColourSchemeTest.java b/test/jalview/schemes/PIDColourSchemeTest.java
index 73b270f..c2f86d6 100644
--- a/test/jalview/schemes/PIDColourSchemeTest.java
+++ b/test/jalview/schemes/PIDColourSchemeTest.java
@@ -4,9 +4,9 @@ import static org.testng.Assert.assertEquals;
 
 import jalview.datamodel.SequenceI;
 import jalview.gui.AlignFrame;
-import jalview.gui.AlignViewport;
 import jalview.io.DataSourceType;
 import jalview.io.FileLoader;
+import jalview.viewmodel.AlignmentViewport;
 
 import java.awt.Color;
 
@@ -42,6 +42,7 @@ public class PIDColourSchemeTest
 
     /*
      * doesn't use column or sequence
+     * we assume consensus residue is computed as upper case
      */
     assertEquals(scheme.findColour('A', 0, null, "A", 0f), white);
     assertEquals(scheme.findColour('A', 0, null, "A", 40f), white);
@@ -54,8 +55,6 @@ public class PIDColourSchemeTest
     assertEquals(scheme.findColour('A', 0, null, "KFV", 100f), white);
 
     assertEquals(scheme.findColour('a', 0, null, "A", 80f), over60);
-    assertEquals(scheme.findColour('A', 0, null, "a", 80f), over60);
-    assertEquals(scheme.findColour('a', 0, null, "a", 80f), over60);
     assertEquals(scheme.findColour('A', 0, null, "AC", 80f), over60);
     assertEquals(scheme.findColour('A', 0, null, "KCA", 80f), over60);
   }
@@ -78,10 +77,26 @@ public class PIDColourSchemeTest
      * first column PID is 50%, or 67% ignoring gaps
      */
     String seqs = ">seq1\nAAAAA\n>seq2\nAAAAA\n>seq3\n-CCCC\n>seq4\nFFFFF\n";
+
+    /*
+     * load data and wait for consensus to be computed
+     */
     AlignFrame af = new FileLoader().LoadFileWaitTillLoaded(seqs,
             DataSourceType.PASTE);
-    AlignViewport viewport = af.getViewport();
+    AlignmentViewport viewport = af.getViewport();
     viewport.setIgnoreGapsConsensus(false, af.alignPanel);
+    while (viewport.getConsensusSeq() == null)
+    {
+      synchronized (this)
+      {
+        try
+        {
+          wait(50);
+        } catch (InterruptedException e)
+        {
+        }
+      }
+    }
     af.changeColour_actionPerformed(JalviewColourScheme.PID.toString());
 
     SequenceI seq = viewport.getAlignment().getSequenceAt(0);