test/jalview/analysis/AAFrequencyTest.java

   1 package jalview.analysis;
   2
   3 import static org.testng.AssertJUnit.assertEquals;
   4 import static org.testng.AssertJUnit.assertNull;
   5
   6 import jalview.datamodel.Sequence;
   7 import jalview.datamodel.SequenceI;
   8
   9 import java.util.Hashtable;
  10
  11 import org.testng.annotations.Test;
  12
  13 public class AAFrequencyTest
  14 {
  15   private static final String C = AAFrequency.MAXCOUNT;
  16
  17   private static final String R = AAFrequency.MAXRESIDUE;
  18
  19   private static final String G = AAFrequency.PID_GAPS;
  20
  21   private static final String N = AAFrequency.PID_NOGAPS;
  22
  23   private static final String P = AAFrequency.PROFILE;
  24
  25   @Test(groups = { "Functional" })
  26   public void testCalculate_noProfile()
  27   {
  28     SequenceI seq1 = new Sequence("Seq1", "CAGT");
  29     SequenceI seq2 = new Sequence("Seq2", "CACT");
  30     SequenceI seq3 = new Sequence("Seq3", "C--G");
  31     SequenceI seq4 = new Sequence("Seq4", "CA-t");
  32     SequenceI[] seqs = new SequenceI[] { seq1, seq2, seq3, seq4 };
  33     Hashtable[] result = new Hashtable[seq1.getLength()];
  34
  35     AAFrequency.calculate(seqs, 0, seq1.getLength(), result, false);
  36
  37     // col 0 is 100% C
  38     Hashtable col = result[0];
  39     assertEquals(100f, (Float) col.get(G), 0.0001f);
  40     assertEquals(100f, (Float) col.get(N), 0.0001f);
  41     assertEquals(4, col.get(C));
  42     assertEquals("C", col.get(R));
  43     assertNull(col.get(P));
  44
  45     // col 1 is 75% A
  46     col = result[1];
  47     assertEquals(75f, (Float) col.get(G), 0.0001f);
  48     assertEquals(100f, (Float) col.get(N), 0.0001f);
  49     assertEquals(3, col.get(C));
  50     assertEquals("A", col.get(R));
  51
  52     // col 2 is 50% G 50% C or 25/25 counting gaps
  53     col = result[2];
  54     assertEquals(25f, (Float) col.get(G), 0.0001f);
  55     assertEquals(50f, (Float) col.get(N), 0.0001f);
  56     assertEquals(1, col.get(C));
  57     assertEquals("CG", col.get(R));
  58
  59     // col 3 is 75% T 25% G
  60     col = result[3];
  61     assertEquals(75f, (Float) col.get(G), 0.0001f);
  62     assertEquals(75f, (Float) col.get(N), 0.0001f);
  63     assertEquals(3, col.get(C));
  64     assertEquals("T", col.get(R));
  65   }
  66
  67   @Test(groups = { "Functional" })
  68   public void testCalculate_withProfile()
  69   {
  70     SequenceI seq1 = new Sequence("Seq1", "CAGT");
  71     SequenceI seq2 = new Sequence("Seq2", "CACT");
  72     SequenceI seq3 = new Sequence("Seq3", "C--G");
  73     SequenceI seq4 = new Sequence("Seq4", "CA-t");
  74     SequenceI[] seqs = new SequenceI[] { seq1, seq2, seq3, seq4 };
  75     Hashtable[] result = new Hashtable[seq1.getLength()];
  76
  77     AAFrequency.calculate(seqs, 0, seq1.getLength(), result, true);
  78     int[][] profile = (int[][]) result[0].get(P);
  79     assertEquals(4, profile[0]['C']);
  80     assertEquals(4, profile[1][0]); // no of seqs
  81     assertEquals(4, profile[1][1]); // nongapped in column
  82
  83     profile = (int[][]) result[1].get(P);
  84     assertEquals(3, profile[0]['A']);
  85     assertEquals(4, profile[1][0]);
  86     assertEquals(3, profile[1][1]);
  87
  88     profile = (int[][]) result[2].get(P);
  89     assertEquals(1, profile[0]['G']);
  90     assertEquals(1, profile[0]['C']);
  91     assertEquals(4, profile[1][0]);
  92     assertEquals(2, profile[1][1]);
  93
  94     profile = (int[][]) result[3].get(P);
  95     assertEquals(3, profile[0]['T']);
  96     assertEquals(1, profile[0]['G']);
  97     assertEquals(4, profile[1][0]);
  98     assertEquals(4, profile[1][1]);
  99   }
 100
 101   @Test(groups = { "Functional" })
 102   public void testCalculate_withProfileTiming()
 103   {
 104     SequenceI seq1 = new Sequence("Seq1", "CAGT");
 105     SequenceI seq2 = new Sequence("Seq2", "CACT");
 106     SequenceI seq3 = new Sequence("Seq3", "C--G");
 107     SequenceI seq4 = new Sequence("Seq4", "CA-t");
 108     SequenceI[] seqs = new SequenceI[] { seq1, seq2, seq3, seq4 };
 109     Hashtable[] result = new Hashtable[seq1.getLength()];
 110
 111     // ensure class loaded and initialized
 112     AAFrequency.calculate(seqs, 0, seq1.getLength(), result, true);
 113     int reps = 100000;
 114     long start = System.currentTimeMillis();
 115     for (int i = 0; i < reps; i++)
 116     {
 117       AAFrequency.calculate(seqs, 0, seq1.getLength(), result, true);
 118     }
 119     System.out.println(System.currentTimeMillis() - start);
 120   }
 121
 122   @Test(groups = { "Functional" })
 123   public void testGetPercentageFormat()
 124   {
 125     assertNull(AAFrequency.getPercentageFormat(0));
 126     assertNull(AAFrequency.getPercentageFormat(99));
 127     assertEquals("%3.1f", AAFrequency.getPercentageFormat(100).toString());
 128     assertEquals("%3.1f", AAFrequency.getPercentageFormat(999).toString());
 129     assertEquals("%3.2f", AAFrequency.getPercentageFormat(1000).toString());
 130     assertEquals("%3.2f", AAFrequency.getPercentageFormat(9999).toString());
 131   }
 132 }