test/jalview/io/AnnotatedPDBFileInputTest.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.io;
  22
  23 import static org.testng.AssertJUnit.assertEquals;
  24 import static org.testng.AssertJUnit.assertNotNull;
  25 import static org.testng.AssertJUnit.assertTrue;
  26
  27 import jalview.bin.Cache;
  28 import jalview.datamodel.AlignmentAnnotation;
  29 import jalview.datamodel.AlignmentI;
  30 import jalview.datamodel.PDBEntry;
  31 import jalview.datamodel.SequenceFeature;
  32 import jalview.datamodel.SequenceI;
  33 import jalview.datamodel.features.SequenceFeatures;
  34 import jalview.gui.AlignFrame;
  35 import jalview.gui.JvOptionPane;
  36 import jalview.structure.StructureImportSettings;
  37 import jalview.structure.StructureImportSettings.StructureParser;
  38
  39 import java.io.File;
  40 import java.util.List;
  41
  42 import org.junit.Assert;
  43 import org.testng.annotations.AfterClass;
  44 import org.testng.annotations.BeforeClass;
  45 import org.testng.annotations.BeforeMethod;
  46 import org.testng.annotations.Test;
  47
  48 public class AnnotatedPDBFileInputTest
  49 {
  50
  51   @BeforeClass(alwaysRun = true)
  52   public void setUpJvOptionPane()
  53   {
  54     JvOptionPane.setInteractiveMode(false);
  55     JvOptionPane.setMockResponse(JvOptionPane.CANCEL_OPTION);
  56   }
  57
  58   AlignmentI al;
  59
  60   String pdbId;
  61
  62   /**
  63    * Ensure 'process secondary structure from PDB and add annotations' are set
  64    * in preferences, and load PDB example file 1gaq
  65    *
  66    * @throws Exception
  67    */
  68   @BeforeMethod(alwaysRun = true)
  69   public void setup() throws Exception
  70   {
  71     Cache.applicationProperties.setProperty("STRUCT_FROM_PDB",
  72             Boolean.TRUE.toString());
  73     Cache.applicationProperties.setProperty("ADD_SS_ANN",
  74             Boolean.TRUE.toString());
  75     FileLoader loader = new FileLoader(false);
  76     AlignFrame af = loader.LoadFileWaitTillLoaded("examples/1gaq.txt",
  77             DataSourceType.FILE);
  78     al = af.getViewport().getAlignment();
  79     pdbId = al.getSequenceAt(0).getDatasetSequence().getAllPDBEntries()
  80             .get(0).getId();
  81     StructureImportSettings.setDefaultStructureFileFormat("PDB");
  82     // StructureImportSettings
  83     // .setDefaultPDBFileParser(StructureParser.JALVIEW_PARSER);
  84   }
  85
  86   @Test(groups = { "Functional" })
  87   public void checkNoDuplicates()
  88   {
  89     // not strictly a requirement, but strange things may happen if multiple
  90     // instances of the same annotation are placed in the alignment annotation
  91     // vector
  92     assertNotNull(al.getAlignmentAnnotation());
  93     // verify that all sequence annotation is doubly referenced
  94     AlignmentAnnotation[] avec = al.getAlignmentAnnotation();
  95     for (int p = 0; p < avec.length; p++)
  96     {
  97       for (int q = p + 1; q < avec.length; q++)
  98       {
  99         assertTrue("Found a duplicate annotation row " + avec[p].label,
 100                 avec[p] != avec[q]);
 101       }
 102     }
 103   }
 104
 105   @Test(groups = { "Functional" }, enabled = false)
 106   public void checkPDBannotationSource()
 107   {
 108     Assert.fail(
 109             "This test is incorrect - does not verify that JmolParser's annotation rows can be recognised as generated by the Jmol parser.");
 110     for (SequenceI asq : al.getSequences())
 111     {
 112       for (AlignmentAnnotation aa : asq.getAnnotation())
 113       {
 114
 115         System.out.println("CalcId: " + aa.getCalcId());
 116         if (StructureImportSettings.getDefaultPDBFileParser()
 117                 .equals(StructureParser.JALVIEW_PARSER))
 118         {
 119           assertTrue(mc_view.PDBfile.isCalcIdForFile(aa, pdbId));
 120         }
 121       }
 122     }
 123   }
 124
 125   /**
 126    * Check sequence features have been added
 127    */
 128   @Test(groups = { "Functional" })
 129   public void checkPDBSequenceFeatures()
 130   {
 131     /*
 132      * 1GAQ/A
 133      */
 134     List<SequenceFeature> sf = al.getSequenceAt(0).getSequenceFeatures();
 135     SequenceFeatures.sortFeatures(sf, true);
 136     assertEquals(296, sf.size());
 137     assertEquals("RESNUM", sf.get(0).getType());
 138     assertEquals("GLU:  19  1gaqA", sf.get(0).getDescription());
 139     assertEquals("RESNUM", sf.get(295).getType());
 140     assertEquals("TYR: 314  1gaqA", sf.get(295).getDescription());
 141
 142     /*
 143      * 1GAQ/B
 144      */
 145     sf = al.getSequenceAt(1).getSequenceFeatures();
 146     SequenceFeatures.sortFeatures(sf, true);
 147     assertEquals(98, sf.size());
 148     assertEquals("RESNUM", sf.get(0).getType());
 149     assertEquals("ALA:   1  1gaqB", sf.get(0).getDescription());
 150     assertEquals("RESNUM", sf.get(97).getType());
 151     assertEquals("ALA:  98  1gaqB", sf.get(97).getDescription());
 152
 153     /*
 154      * 1GAQ/C
 155      */
 156     sf = al.getSequenceAt(2).getSequenceFeatures();
 157     SequenceFeatures.sortFeatures(sf, true);
 158     assertEquals(296, sf.size());
 159     assertEquals("RESNUM", sf.get(0).getType());
 160     assertEquals("GLU:  19  1gaqC", sf.get(0).getDescription());
 161     assertEquals("RESNUM", sf.get(295).getType());
 162     assertEquals("TYR: 314  1gaqC", sf.get(295).getDescription());
 163   }
 164
 165   @Test(groups = { "Functional" })
 166   public void checkAnnotationWiring()
 167   {
 168     assertTrue(al.getAlignmentAnnotation() != null);
 169     // verify that all sequence annotation is doubly referenced
 170     for (AlignmentAnnotation aa : al.getAlignmentAnnotation())
 171     {
 172       if (aa.sequenceRef != null)
 173       {
 174         assertTrue(al.getSequences().contains(aa.sequenceRef));
 175         assertNotNull(aa.sequenceRef.getAnnotation());
 176         boolean found = false;
 177         for (AlignmentAnnotation sqan : aa.sequenceRef.getAnnotation())
 178         {
 179           if (sqan == aa)
 180           {
 181             found = true;
 182             break;
 183           }
 184         }
 185         assertTrue("Couldn't find sequence associated annotation "
 186                 + aa.label
 187                 + " on the sequence it is associated with.\nSequence associated editing will fail.",
 188                 found);
 189       }
 190     }
 191   }
 192
 193   /**
 194    * @throws java.lang.Exception
 195    */
 196   @BeforeClass(alwaysRun = true)
 197   public static void setUpBeforeClass() throws Exception
 198   {
 199     jalview.bin.Jalview
 200             .main(new String[]
 201             { "-props", "test/jalview/io/testProps.jvprops" });
 202   }
 203
 204   /**
 205    * @throws java.lang.Exception
 206    */
 207   @AfterClass(alwaysRun = true)
 208   public static void tearDownAfterClass() throws Exception
 209   {
 210     jalview.gui.Desktop.instance.closeAll_actionPerformed(null);
 211
 212   }
 213
 214   @Test(groups = { "Functional" })
 215   public void testJalviewProjectRelocationAnnotation() throws Exception
 216   {
 217
 218     String inFile = "examples/1gaq.txt";
 219     String tfile = File.createTempFile("JalviewTest", ".jvp")
 220             .getAbsolutePath();
 221     AlignFrame af = new jalview.io.FileLoader()
 222             .LoadFileWaitTillLoaded(inFile, DataSourceType.FILE);
 223     assertTrue("Didn't read input file " + inFile, af != null);
 224     af.saveAlignment(tfile, FileFormat.Jalview);
 225     assertTrue("Failed to store as a project.",
 226             af.isSaveAlignmentSuccessful());
 227     af.closeMenuItem_actionPerformed(true);
 228     af = null;
 229     af = new jalview.io.FileLoader().LoadFileWaitTillLoaded(tfile,
 230             DataSourceType.FILE);
 231     assertTrue("Failed to import new project", af != null);
 232     for (SequenceI asq : af.getViewport().getAlignment().getSequences())
 233     {
 234       SequenceI sq = asq;
 235       while (sq.getDatasetSequence() != null)
 236       {
 237         sq = sq.getDatasetSequence();
 238       }
 239       assertNotNull(sq.getAllPDBEntries());
 240       assertEquals("Expected only one PDB ID", 1,
 241               sq.getAllPDBEntries().size());
 242       for (PDBEntry pdbentry : sq.getAllPDBEntries())
 243       {
 244         System.err.println(
 245                 "PDB Entry " + pdbentry.getId() + " " + pdbentry.getFile());
 246         boolean exists = false, found = false;
 247         for (AlignmentAnnotation ana : sq.getAnnotation())
 248         {
 249           System.err.println("CalcId " + ana.getCalcId());
 250           if (ana.getCalcId() != null
 251                   && mc_view.PDBfile.isCalcIdHandled(ana.getCalcId()))
 252           {
 253             exists = true;
 254             if (mc_view.PDBfile.isCalcIdForFile(ana, pdbentry.getId()))
 255             {
 256               found = true;
 257             }
 258           }
 259         }
 260         if (exists)
 261         {
 262           assertTrue(
 263                   "Couldn't find any annotation for " + pdbentry.getId()
 264                           + " (file handle " + pdbentry.getFile() + ")",
 265                   found);
 266         }
 267       }
 268     }
 269   }
 270 }