test/jalview/gui/StructureViewerTest.java

   1 package jalview.gui;
   2
   3 import static org.testng.Assert.assertEquals;
   4 import static org.testng.Assert.assertFalse;
   5 import static org.testng.Assert.assertNull;
   6 import static org.testng.Assert.assertSame;
   7 import static org.testng.Assert.assertTrue;
   8
   9 import jalview.datamodel.PDBEntry;
  10 import jalview.datamodel.PDBEntry.Type;
  11 import jalview.datamodel.Sequence;
  12 import jalview.datamodel.SequenceI;
  13
  14 import java.util.Map;
  15
  16 import org.testng.annotations.BeforeClass;
  17 import org.testng.annotations.Test;
  18
  19 public class StructureViewerTest
  20 {
  21
  22   @BeforeClass(alwaysRun = true)
  23   public void setUpJvOptionPane()
  24   {
  25     JvOptionPane.setInteractiveMode(false);
  26     JvOptionPane.setMockResponse(JvOptionPane.CANCEL_OPTION);
  27   }
  28
  29   @Test(groups = "Functional")
  30   public void testGetSequencesForPdbs()
  31   {
  32     assertNull(StructureViewer.getSequencesForPdbs(null, null));
  33
  34     PDBEntry pdbe1 = new PDBEntry("1A70", "A", Type.PDB, "path1");
  35     PDBEntry pdbe2 = new PDBEntry("3A6S", "A", Type.PDB, "path2");
  36     PDBEntry pdbe3 = new PDBEntry("1A70", "B", Type.PDB, "path1");
  37     PDBEntry pdbe4 = new PDBEntry("1GAQ", "A", Type.PDB, null);
  38     PDBEntry pdbe5 = new PDBEntry("3A6S", "B", Type.PDB, "path2");
  39     PDBEntry pdbe6 = new PDBEntry("1GAQ", "B", Type.PDB, null);
  40     PDBEntry[] pdbs = new PDBEntry[] { pdbe1, pdbe2, pdbe3, pdbe4, pdbe5,
  41         pdbe6 };
  42
  43     /*
  44      * seq1 ... seq6 associated with pdbe1 ... pdbe6
  45      */
  46     SequenceI[] seqs = new SequenceI[pdbs.length];
  47     for (int i = 0; i < seqs.length; i++)
  48     {
  49       seqs[i] = new Sequence("Seq" + i, "abc");
  50     }
  51
  52     /*
  53      * pdbe3/5/6 should get removed as having a duplicate file path
  54      */
  55     Map<PDBEntry, SequenceI[]> uniques = StructureViewer
  56             .getSequencesForPdbs(pdbs, seqs);
  57     assertTrue(uniques.containsKey(pdbe1));
  58     assertTrue(uniques.containsKey(pdbe2));
  59     assertFalse(uniques.containsKey(pdbe3));
  60     assertTrue(uniques.containsKey(pdbe4));
  61     assertFalse(uniques.containsKey(pdbe5));
  62     assertFalse(uniques.containsKey(pdbe6));
  63
  64     // 1A70 associates with seq1 and seq3
  65     SequenceI[] ss = uniques.get(pdbe1);
  66     assertEquals(ss.length, 2);
  67     assertSame(seqs[0], ss[0]);
  68     assertSame(seqs[2], ss[1]);
  69
  70     // 3A6S has seq2 and seq5
  71     ss = uniques.get(pdbe2);
  72     assertEquals(ss.length, 2);
  73     assertSame(seqs[1], ss[0]);
  74     assertSame(seqs[4], ss[1]);
  75
  76     // 1GAQ has seq4 and seq6
  77     ss = uniques.get(pdbe4);
  78     assertEquals(ss.length, 2);
  79     assertSame(seqs[3], ss[0]);
  80     assertSame(seqs[5], ss[1]);
  81   }
  82 }