JAL-1705 refactored/utility methods to detect e.g. 'PhenCode_variation'

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

diff --git a/test/jalview/ext/ensembl/EnsemblSeqProxyTest.java b/test/jalview/ext/ensembl/EnsemblSeqProxyTest.java
index 10ecfe0..71f0212 100644 (file)
--- a/test/jalview/ext/ensembl/EnsemblSeqProxyTest.java
+++ b/test/jalview/ext/ensembl/EnsemblSeqProxyTest.java
@@ -1,11 +1,19 @@
 package jalview.ext.ensembl;
 
+import static org.testng.AssertJUnit.assertEquals;
+import static org.testng.AssertJUnit.assertFalse;
+import static org.testng.AssertJUnit.assertTrue;
+import static org.testng.internal.junit.ArrayAsserts.assertArrayEquals;
+
 import jalview.datamodel.Alignment;
 import jalview.datamodel.AlignmentI;
+import jalview.datamodel.SequenceFeature;
 import jalview.datamodel.SequenceI;
 import jalview.io.AppletFormatAdapter;
 import jalview.io.FastaFile;
 import jalview.io.FileParse;
+import jalview.io.gff.SequenceOntologyFactory;
+import jalview.io.gff.SequenceOntologyLite;
 
 import java.lang.reflect.Method;
 import java.net.MalformedURLException;
@@ -14,6 +22,8 @@ import java.util.Arrays;
 import java.util.List;
 
 import org.testng.Assert;
+import org.testng.annotations.AfterClass;
+import org.testng.annotations.BeforeClass;
 import org.testng.annotations.DataProvider;
 import org.testng.annotations.Test;
 
@@ -99,20 +109,16 @@ public class EnsemblSeqProxyTest
                   + "NRDQIIFMVGRGYLSPDLSKVRSNCPKAMKRLMAECLKKKRDERPLFPQILASIELLARS\n"
                   + "LPKIHRSASEPSLNRAGFQTEDFSLYACASPKTPIQAGGYGAFPVH" } };
 
-  @DataProvider(name = "queries")
-  public Object[][] createQueryData(Method m)
+  @BeforeClass
+  public void setUp()
   {
-    return new Object[][] { { "CCDS5863.1" }, { "ENSP00000288602" } };
+    SequenceOntologyFactory.setInstance(new SequenceOntologyLite());
   }
 
-  @Test(dataProvider = "queries")
-  public void testIsValidReference(String query) throws Exception
+  @AfterClass
+  public void tearDown()
   {
-    EnsemblSequenceFetcher esq = new EnsemblProtein();
-    Assert.assertTrue(esq.isValidReference(query),
-            "Expected reference string " + query
-                    + " to be valid for regex "
-                    + esq.getAccessionValidator().toString());
+    SequenceOntologyFactory.setInstance(null);
   }
 
   @DataProvider(name = "ens_seqs")
@@ -208,5 +214,102 @@ public class EnsemblSeqProxyTest
             + (isAvailable ? "UP!"
                     : "DOWN or unreachable ******************* BAD!"));
   }
-  // todo lots of tests
+
+  @Test(groups = "Functional")
+  public void getGenomicRangesFromFeatures()
+  {
+
+  }
+
+  @Test(groups = "Functional")
+  public void testIsTranscriptIdentifier()
+  {
+    EnsemblSeqProxy testee = new EnsemblGene();
+    assertFalse(testee.isTranscriptIdentifier(null));
+    assertFalse(testee.isTranscriptIdentifier(""));
+    assertFalse(testee.isTranscriptIdentifier("ENSG00000012345"));
+    assertTrue(testee.isTranscriptIdentifier("ENST00000012345"));
+    assertTrue(testee.isTranscriptIdentifier("ENSMUST00000012345"));
+    assertFalse(testee.isTranscriptIdentifier("enst00000012345"));
+    assertFalse(testee.isTranscriptIdentifier("ENST000000123456"));
+    assertFalse(testee.isTranscriptIdentifier("ENST0000001234"));
+  }
+
+  @Test(groups = "Functional")
+  public void testIsGeneIdentifier()
+  {
+    EnsemblSeqProxy testee = new EnsemblGene();
+    assertFalse(testee.isGeneIdentifier(null));
+    assertFalse(testee.isGeneIdentifier(""));
+    assertFalse(testee.isGeneIdentifier("ENST00000012345"));
+    assertTrue(testee.isGeneIdentifier("ENSG00000012345"));
+    assertTrue(testee.isGeneIdentifier("ENSMUSG00000012345"));
+    assertFalse(testee.isGeneIdentifier("ensg00000012345"));
+    assertFalse(testee.isGeneIdentifier("ENSG000000123456"));
+    assertFalse(testee.isGeneIdentifier("ENSG0000001234"));
+  }
+
+  /**
+   * Test the method that appends a single allele's reverse complement to a
+   * string buffer
+   */
+  @Test(groups = "Functional")
+  public void testReverseComplementAllele()
+  {
+    StringBuilder sb = new StringBuilder();
+    EnsemblSeqProxy.reverseComplementAllele(sb, "G"); // comp=C
+    EnsemblSeqProxy.reverseComplementAllele(sb, "g"); // comp=c
+    EnsemblSeqProxy.reverseComplementAllele(sb, "C"); // comp=G
+    EnsemblSeqProxy.reverseComplementAllele(sb, "T"); // comp=A
+    EnsemblSeqProxy.reverseComplementAllele(sb, "A"); // comp=T
+    assertEquals("C,c,G,A,T", sb.toString());
+
+    sb = new StringBuilder();
+    EnsemblSeqProxy.reverseComplementAllele(sb, "-GATt"); // revcomp=aATC-
+    EnsemblSeqProxy.reverseComplementAllele(sb, "hgmd_mutation");
+    EnsemblSeqProxy.reverseComplementAllele(sb, "PhenCode_variation");
+    assertEquals("aATC-,hgmd_mutation,PhenCode_variation", sb.toString());
+  }
+
+  /**
+   * Test the method that computes the reverse complement of the alleles in a
+   * sequence_variant feature
+   */
+  @Test(groups = "Functional")
+  public void testReverseComplementAlleles()
+  {
+    String alleles = "C,G,-TAC,HGMD_MUTATION,gac";
+    SequenceFeature sf = new SequenceFeature("sequence_variant", alleles,
+            1, 2, 0f, null);
+    sf.setValue("alleles", alleles);
+    sf.setAttributes("x=y,z;alleles=" + alleles + ";a=b,c");
+
+    EnsemblSeqProxy.reverseComplementAlleles(sf);
+    String revcomp = "G,C,GTA-,HGMD_MUTATION,gtc";
+    // verify description is updated with reverse complement
+    assertEquals(revcomp, sf.getDescription());
+    // verify alleles attribute is updated with reverse complement
+    assertEquals(revcomp, sf.getValue("alleles"));
+    // verify attributes string is updated with reverse complement
+    assertEquals("x=y,z;alleles=" + revcomp + ";a=b,c", sf.getAttributes());
+  }
+
+  @Test(groups = "Functional")
+  public void testSortFeatures()
+  {
+    SequenceFeature sf1 = new SequenceFeature("", "", 10, 15, 0f, null);
+    SequenceFeature sf2 = new SequenceFeature("", "", 8, 12, 0f, null);
+    SequenceFeature sf3 = new SequenceFeature("", "", 8, 13, 0f, null);
+    SequenceFeature sf4 = new SequenceFeature("", "", 11, 11, 0f, null);
+    SequenceFeature[] sfs = new SequenceFeature[] { sf1, sf2, sf3, sf4 };
+
+    // sort by start position ascending (forward strand)
+    // sf2 and sf3 tie and should not be reordered by sorting
+    EnsemblSeqProxy.sortFeatures(sfs, true);
+    assertArrayEquals(new SequenceFeature[] { sf2, sf3, sf1, sf4 }, sfs);
+
+    // sort by end position descending (reverse strand)
+    EnsemblSeqProxy.sortFeatures(sfs, false);
+    assertArrayEquals(new SequenceFeature[] { sf1, sf3, sf2, sf4 }, sfs);
+  }
 }
\ No newline at end of file