Merge branch 'develop' into trialMerge
[jalview.git] / test / jalview / structures / models / AAStructureBindingModelTest.java
1 /*
2  * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
3  * Copyright (C) $$Year-Rel$$ The Jalview Authors
4  * 
5  * This file is part of Jalview.
6  * 
7  * Jalview is free software: you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License 
9  * as published by the Free Software Foundation, either version 3
10  * of the License, or (at your option) any later version.
11  *  
12  * Jalview is distributed in the hope that it will be useful, but 
13  * WITHOUT ANY WARRANTY; without even the implied warranty 
14  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
15  * PURPOSE.  See the GNU General Public License for more details.
16  * 
17  * You should have received a copy of the GNU General Public License
18  * along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
19  * The Jalview Authors are detailed in the 'AUTHORS' file.
20  */
21 package jalview.structures.models;
22
23 import static org.testng.AssertJUnit.assertEquals;
24 import static org.testng.AssertJUnit.assertFalse;
25 import static org.testng.AssertJUnit.assertTrue;
26
27 import jalview.datamodel.Alignment;
28 import jalview.datamodel.AlignmentI;
29 import jalview.datamodel.PDBEntry;
30 import jalview.datamodel.PDBEntry.Type;
31 import jalview.datamodel.Sequence;
32 import jalview.datamodel.SequenceI;
33 import jalview.gui.JvOptionPane;
34 import jalview.io.DataSourceType;
35 import jalview.structure.AtomSpec;
36 import jalview.structure.StructureSelectionManager;
37 import jalview.structures.models.AAStructureBindingModel.SuperposeData;
38
39 import java.util.Arrays;
40 import java.util.List;
41
42 import org.testng.annotations.BeforeClass;
43 import org.testng.annotations.BeforeMethod;
44 import org.testng.annotations.Test;
45
46 /**
47  * Unit tests for non-abstract methods of abstract base class
48  * 
49  * @author gmcarstairs
50  *
51  */
52 public class AAStructureBindingModelTest
53 {
54
55   @BeforeClass(alwaysRun = true)
56   public void setUpJvOptionPane()
57   {
58     JvOptionPane.setInteractiveMode(false);
59     JvOptionPane.setMockResponse(JvOptionPane.CANCEL_OPTION);
60   }
61
62   /*
63    * Scenario: Jalview has 4 sequences, corresponding to 1YCS (chains A and B), 3A6S|B, 1OOT|A
64    */
65   private static final String PDB_1 = "HEADER    COMPLEX (ANTI-ONCOGENE/ANKYRIN REPEATS) 30-SEP-96   1YCS              \n"
66           + "ATOM      2  CA  VAL A  97      24.134   4.926  45.821  1.00 47.43           C  \n"
67           + "ATOM      9  CA  PRO A  98      25.135   8.584  46.217  1.00 41.60           C  \n"
68           + "ATOM     16  CA  SER A  99      28.243   9.596  44.271  1.00 39.63           C  \n"
69           + "ATOM     22  CA  GLN A 100      31.488  10.133  46.156  1.00 35.60           C  \n"
70           // artificial jump in residue numbering to prove it is correctly
71           // mapped:
72           + "ATOM     31  CA  LYS A 102      33.323  11.587  43.115  1.00 41.69           C  \n"
73           + "ATOM   1857  CA  GLU B 374       9.193 -16.005  95.870  1.00 54.22           C  \n"
74           + "ATOM   1866  CA  ILE B 375       7.101 -14.921  92.847  1.00 46.82           C  \n"
75           + "ATOM   1874  CA  VAL B 376      10.251 -13.625  91.155  1.00 47.80           C  \n"
76           + "ATOM   1881  CA  LYS B 377      11.767 -17.068  91.763  1.00 50.21           C  \n"
77           + "ATOM   1890  CA  PHE B 378       8.665 -18.948  90.632  1.00 44.85           C  \n";
78
79   private static final String PDB_2 = "HEADER    HYDROLASE                               09-SEP-09   3A6S              \n"
80           + "ATOM      2  CA  MET B   1      15.366 -11.648  24.854  1.00 32.05           C  \n"
81           + "ATOM     10  CA  LYS B   2      16.846  -9.215  22.340  1.00 25.68           C  \n"
82           + "ATOM     19  CA  LYS B   3      15.412  -6.335  20.343  1.00 19.42           C  \n"
83           + "ATOM     28  CA  LEU B   4      15.629  -5.719  16.616  1.00 15.49           C  \n"
84           + "ATOM     36  CA  GLN B   5      14.412  -2.295  15.567  1.00 12.19           C  \n";
85
86   private static final String PDB_3 = "HEADER    STRUCTURAL GENOMICS                     04-MAR-03   1OOT              \n"
87           + "ATOM      2  CA  SER A   7      29.427   3.330  -6.578  1.00 32.50           C  \n"
88           + "ATOM      8  CA  PRO A   8      29.975   3.340  -2.797  1.00 17.62           C  \n"
89           + "ATOM     16  CA ALYS A   9      26.958   3.024  -0.410  0.50  8.78           C  \n"
90           + "ATOM     33  CA  ALA A  10      26.790   4.320   3.172  1.00 11.98           C  \n"
91           + "ATOM     39  CA AVAL A  12      24.424   3.853   6.106  0.50 13.83           C  \n";
92
93   AAStructureBindingModel testee;
94
95   AlignmentI al = null;
96
97   /**
98    * Set up test conditions with three aligned sequences,
99    */
100   @BeforeMethod(alwaysRun = true)
101   public void setUp()
102   {
103     SequenceI seq1a = new Sequence("1YCS|A", "-VPSQK");
104     SequenceI seq1b = new Sequence("1YCS|B", "EIVKF-");
105     SequenceI seq2 = new Sequence("3A6S", "MK-KLQ");
106     SequenceI seq3 = new Sequence("1OOT", "SPK-AV");
107     al = new Alignment(new SequenceI[] { seq1a, seq1b, seq2, seq3 });
108     al.setDataset(null);
109
110     /*
111      * give pdb files the name generated by Jalview for PASTE source
112      */
113     PDBEntry[] pdbFiles = new PDBEntry[3];
114     pdbFiles[0] = new PDBEntry("1YCS", "A", Type.PDB, "INLINE1YCS");
115     pdbFiles[1] = new PDBEntry("3A6S", "B", Type.PDB, "INLINE3A6S");
116     pdbFiles[2] = new PDBEntry("1OOT", "A", Type.PDB, "INLINE1OOT");
117     SequenceI[][] seqs = new SequenceI[3][];
118     seqs[0] = new SequenceI[] { seq1a, seq1b };
119     seqs[1] = new SequenceI[] { seq2 };
120     seqs[2] = new SequenceI[] { seq3 };
121     StructureSelectionManager ssm = new StructureSelectionManager();
122
123     ssm.setMapping(new SequenceI[] { seq1a, seq1b }, null, PDB_1,
124             DataSourceType.PASTE);
125     ssm.setMapping(new SequenceI[] { seq2 }, null, PDB_2,
126             DataSourceType.PASTE);
127     ssm.setMapping(new SequenceI[] { seq3 }, null, PDB_3,
128             DataSourceType.PASTE);
129
130     testee = new AAStructureBindingModel(ssm, pdbFiles, seqs, null)
131     {
132       @Override
133       public String[] getPdbFile()
134       {
135         return new String[] { "INLINE1YCS", "INLINE3A6S", "INLINE1OOT" };
136       }
137
138       @Override
139       public void updateColours(Object source)
140       {
141       }
142
143       @Override
144       public void releaseReferences(Object svl)
145       {
146       }
147
148       @Override
149       public void highlightAtoms(List<AtomSpec> atoms)
150       {
151       }
152
153       @Override
154       public List<String> getChainNames()
155       {
156         return null;
157       }
158     };
159   }
160
161   /**
162    * Verify that the method determines that columns 2, 5 and 6 of the alignment
163    * are alignable in structure
164    */
165   @Test(groups = { "Functional" })
166   public void testFindSuperposableResidues()
167   {
168     /*
169      * create a data bean to hold data per structure file
170      */
171     SuperposeData[] structs = new SuperposeData[testee.getPdbFile().length];
172     for (int i = 0; i < structs.length; i++)
173     {
174       structs[i] = testee.new SuperposeData(al.getWidth());
175     }
176     /*
177      * initialise array of 'superposable columns' to true (would be false for
178      * hidden columns)
179      */
180     boolean[] matched = new boolean[al.getWidth()];
181     Arrays.fill(matched, true);
182
183     int refStructure = testee
184             .findSuperposableResidues(al, matched, structs);
185
186     assertEquals(0, refStructure);
187
188     /*
189      * only ungapped, structure-mapped columns are superposable
190      */
191     assertFalse(matched[0]); // gap in first sequence
192     assertTrue(matched[1]);
193     assertFalse(matched[2]); // gap in third sequence
194     assertFalse(matched[3]); // gap in fourth sequence
195     assertTrue(matched[4]);
196     assertTrue(matched[5]); // gap in second sequence
197
198     assertEquals("1YCS", structs[0].pdbId);
199     assertEquals("3A6S", structs[1].pdbId);
200     assertEquals("1OOT", structs[2].pdbId);
201     assertEquals("A", structs[0].chain); // ? struct has chains A _and_ B
202     assertEquals("B", structs[1].chain);
203     assertEquals("A", structs[2].chain);
204     // the 0's for unsuperposable positions propagate down the columns:
205     assertEquals("[0, 97, 98, 99, 100, 102]",
206             Arrays.toString(structs[0].pdbResNo));
207     assertEquals("[0, 2, 0, 3, 4, 5]", Arrays.toString(structs[1].pdbResNo));
208     assertEquals("[0, 8, 0, 0, 10, 12]",
209             Arrays.toString(structs[2].pdbResNo));
210   }
211
212   @Test(groups = { "Functional" })
213   public void testFindSuperposableResidues_hiddenColumn()
214   {
215     SuperposeData[] structs = new SuperposeData[al.getHeight()];
216     for (int i = 0; i < structs.length; i++)
217     {
218       structs[i] = testee.new SuperposeData(al.getWidth());
219     }
220     /*
221      * initialise array of 'superposable columns' to true (would be false for
222      * hidden columns)
223      */
224     boolean[] matched = new boolean[al.getWidth()];
225     Arrays.fill(matched, true);
226     // treat column 5 of the alignment as hidden
227     matched[4] = false;
228
229     int refStructure = testee
230             .findSuperposableResidues(al, matched, structs);
231
232     assertEquals(0, refStructure);
233
234     // only ungapped, structure-mapped columns are not superposable
235     assertFalse(matched[0]);
236     assertTrue(matched[1]);
237     assertFalse(matched[2]);
238     assertFalse(matched[3]);
239     assertFalse(matched[4]); // superposable, but hidden, column
240     assertTrue(matched[5]);
241   }
242 }