JAL-2360 ColourSchemes holds configured schemes, AlignFrame colour menu

[jalview.git] / test / jalview / schemes / JalviewColourSchemeTest.java

diff --git a/test/jalview/schemes/JalviewColourSchemeTest.java b/test/jalview/schemes/JalviewColourSchemeTest.java
index b5246ba..d9403d2 100644 (file)
--- a/test/jalview/schemes/JalviewColourSchemeTest.java
+++ b/test/jalview/schemes/JalviewColourSchemeTest.java
@@ -1,82 +1,48 @@
 package jalview.schemes;
 
-import static org.testng.Assert.assertNull;
-import static org.testng.Assert.assertSame;
+import static org.testng.Assert.assertEquals;
 import static org.testng.Assert.assertTrue;
 
-import jalview.datamodel.Alignment;
-import jalview.datamodel.AlignmentI;
-import jalview.datamodel.Sequence;
-import jalview.datamodel.SequenceI;
-
 import org.testng.annotations.Test;
 
 public class JalviewColourSchemeTest
 {
   @Test(groups = "Functional")
-  public void testForName()
+  public void testGetSchemeClass()
   {
-    assertSame(JalviewColourScheme.Clustal,
-            JalviewColourScheme.forName("Clustal"));
-    assertSame(JalviewColourScheme.Clustal,
-            JalviewColourScheme.forName("CLUSTAL"));
-    assertSame(JalviewColourScheme.Blosum62,
-            JalviewColourScheme.forName("Blosum62"));
-    assertSame(JalviewColourScheme.PID,
-            JalviewColourScheme.forName("% Identity"));
-    assertSame(JalviewColourScheme.Hydrophobic,
-            JalviewColourScheme.forName("Hydrophobic"));
-    assertSame(JalviewColourScheme.Zappo,
-            JalviewColourScheme.forName("Zappo"));
-    assertSame(JalviewColourScheme.Taylor,
-            JalviewColourScheme.forName("Taylor"));
-    assertSame(JalviewColourScheme.Helix,
-            JalviewColourScheme.forName("Helix Propensity"));
-    assertSame(JalviewColourScheme.Strand,
-            JalviewColourScheme.forName("Strand Propensity"));
-    assertSame(JalviewColourScheme.Turn,
-            JalviewColourScheme.forName("Turn Propensity"));
-    assertSame(JalviewColourScheme.Buried,
-            JalviewColourScheme.forName("Buried Index"));
-    assertSame(JalviewColourScheme.Nucleotide,
-            JalviewColourScheme.forName("Nucleotide"));
-    assertSame(JalviewColourScheme.PurinePyrimidine,
-            JalviewColourScheme.forName("Purine/Pyrimidine"));
-    assertSame(JalviewColourScheme.TCoffee,
-            JalviewColourScheme.forName("T-Coffee Scores"));
-    assertSame(JalviewColourScheme.RNAHelices,
-            JalviewColourScheme.forName("RNA Helices"));
-    // assertSame(JalviewColourScheme.RNAInteraction,
-    // JalviewColourScheme.forName("RNA Interaction type"));
-    assertSame(JalviewColourScheme.UserDefined,
-            JalviewColourScheme.forName("User Defined"));
-
-    assertNull(JalviewColourScheme.forName("PID"));
-    assertNull(JalviewColourScheme.forName(""));
-    assertNull(JalviewColourScheme.forName(null));
+    assertTrue(JalviewColourScheme.Clustal.getSchemeClass() == ClustalxColourScheme.class);
+    assertTrue(JalviewColourScheme.Blosum62.getSchemeClass() == Blosum62ColourScheme.class);
+    assertTrue(JalviewColourScheme.PID.getSchemeClass() == PIDColourScheme.class);
+    assertTrue(JalviewColourScheme.Hydrophobic.getSchemeClass() == HydrophobicColourScheme.class);
+    assertTrue(JalviewColourScheme.Zappo.getSchemeClass() == ZappoColourScheme.class);
+    assertTrue(JalviewColourScheme.Taylor.getSchemeClass() == TaylorColourScheme.class);
+    assertTrue(JalviewColourScheme.Helix.getSchemeClass() == HelixColourScheme.class);
+    assertTrue(JalviewColourScheme.Strand.getSchemeClass() == StrandColourScheme.class);
+    assertTrue(JalviewColourScheme.Turn.getSchemeClass() == TurnColourScheme.class);
+    assertTrue(JalviewColourScheme.Buried.getSchemeClass() == BuriedColourScheme.class);
+    assertTrue(JalviewColourScheme.Nucleotide.getSchemeClass() == NucleotideColourScheme.class);
+    assertTrue(JalviewColourScheme.PurinePyrimidine.getSchemeClass() == PurinePyrimidineColourScheme.class);
+    assertTrue(JalviewColourScheme.TCoffee.getSchemeClass() == TCoffeeColourScheme.class);
+    assertTrue(JalviewColourScheme.RNAHelices.getSchemeClass() == RNAHelicesColour.class);
   }
 
   @Test(groups = "Functional")
-  public void testGetColourScheme()
+  public void testToString()
   {
-    SequenceI seq = new Sequence("Seq1", "abcd");
-    AlignmentI al = new Alignment(new SequenceI[] { seq });
-    assertTrue(JalviewColourScheme.Clustal.getColourScheme(al) instanceof ClustalxColourScheme);
-    assertTrue(JalviewColourScheme.Blosum62.getColourScheme(al) instanceof Blosum62ColourScheme);
-    assertTrue(JalviewColourScheme.PID.getColourScheme(al) instanceof PIDColourScheme);
-    assertTrue(JalviewColourScheme.Hydrophobic.getColourScheme(al) instanceof HydrophobicColourScheme);
-    assertTrue(JalviewColourScheme.Zappo.getColourScheme(al) instanceof ZappoColourScheme);
-    assertTrue(JalviewColourScheme.Taylor.getColourScheme(al) instanceof TaylorColourScheme);
-    assertTrue(JalviewColourScheme.Helix.getColourScheme(al) instanceof HelixColourScheme);
-    assertTrue(JalviewColourScheme.Strand.getColourScheme(al) instanceof StrandColourScheme);
-    assertTrue(JalviewColourScheme.Turn.getColourScheme(al) instanceof TurnColourScheme);
-    assertTrue(JalviewColourScheme.Buried.getColourScheme(al) instanceof BuriedColourScheme);
-    assertTrue(JalviewColourScheme.Nucleotide.getColourScheme(al) instanceof NucleotideColourScheme);
-    assertTrue(JalviewColourScheme.PurinePyrimidine.getColourScheme(al) instanceof PurinePyrimidineColourScheme);
-    assertTrue(JalviewColourScheme.TCoffee.getColourScheme(al) instanceof TCoffeeColourScheme);
-    assertTrue(JalviewColourScheme.RNAHelices.getColourScheme(al) instanceof RNAHelicesColour);
-    // assertTrue(JalviewColourScheme.RNAInteraction.getColourScheme(al)
-    // instanceof RNAInteractionColourScheme);
-    assertTrue(JalviewColourScheme.UserDefined.getColourScheme(al) instanceof UserColourScheme);
+    assertEquals(JalviewColourScheme.Clustal.toString(), "Clustal");
+    assertEquals(JalviewColourScheme.Blosum62.toString(), "Blosum62");
+    assertEquals(JalviewColourScheme.PID.toString(), "% Identity");
+    assertEquals(JalviewColourScheme.Zappo.toString(), "Zappo");
+    assertEquals(JalviewColourScheme.Taylor.toString(), "Taylor");
+    assertEquals(JalviewColourScheme.Hydrophobic.toString(), "Hydrophobic");
+    assertEquals(JalviewColourScheme.Helix.toString(), "Helix Propensity");
+    assertEquals(JalviewColourScheme.Strand.toString(), "Strand Propensity");
+    assertEquals(JalviewColourScheme.Turn.toString(), "Turn Propensity");
+    assertEquals(JalviewColourScheme.Buried.toString(), "Buried Index");
+    assertEquals(JalviewColourScheme.Nucleotide.toString(), "Nucleotide");
+    assertEquals(JalviewColourScheme.PurinePyrimidine.toString(),
+            "Purine/Pyrimidine");
+    assertEquals(JalviewColourScheme.TCoffee.toString(), "T-Coffee Scores");
+    assertEquals(JalviewColourScheme.RNAHelices.toString(), "RNA Helices");
   }
 }