JAL-2783 accept Ensembl query of the form "human:braf"

[jalview.git] / test / jalview / ext / ensembl / SpeciesTest.java

diff --git a/test/jalview/ext/ensembl/SpeciesTest.java b/test/jalview/ext/ensembl/SpeciesTest.java
index 44658e7..2afa159 100644 (file)
--- a/test/jalview/ext/ensembl/SpeciesTest.java
+++ b/test/jalview/ext/ensembl/SpeciesTest.java
@@ -3,7 +3,7 @@ package jalview.ext.ensembl;
 import static org.testng.Assert.assertFalse;
 import static org.testng.Assert.assertTrue;
 
-import java.util.Set;
+import java.util.List;
 
 import org.testng.annotations.Test;
 
@@ -12,18 +12,18 @@ public class SpeciesTest
   @Test
   public void testGetModelOrganisms()
   {
-    Set<Species> models = Species.getModelOrganisms();
-    assertTrue(models.contains(Species.human));
-    assertFalse(models.contains(Species.horse));
+    List<String> models = Species.getModelOrganisms();
+    assertTrue(models.contains(Species.human.name()));
+    assertFalse(models.contains(Species.horse.name()));
     for (Species s : Species.values())
     {
       if (s.isModelOrganism())
       {
-        assertTrue(models.contains(s));
+        assertTrue(models.contains(s.name()));
       }
       else
       {
-        assertFalse(models.contains(s));
+        assertFalse(models.contains(s.name()));
       }
     }
   }