2 * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
3 * Copyright (C) $$Year-Rel$$ The Jalview Authors
5 * This file is part of Jalview.
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.
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.
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.
21 package jalview.controller;
23 import static org.testng.AssertJUnit.assertEquals;
24 import static org.testng.AssertJUnit.assertTrue;
26 import jalview.analysis.Finder;
27 import jalview.api.AlignViewControllerI;
28 import jalview.datamodel.SearchResults;
29 import jalview.datamodel.SearchResultsI;
30 import jalview.datamodel.Sequence;
31 import jalview.datamodel.SequenceFeature;
32 import jalview.datamodel.SequenceGroup;
33 import jalview.datamodel.SequenceI;
34 import jalview.gui.AlignFrame;
35 import jalview.gui.JvOptionPane;
36 import jalview.io.FileLoader;
37 import jalview.io.FormatAdapter;
39 import java.util.Arrays;
40 import java.util.BitSet;
42 import org.testng.annotations.BeforeClass;
43 import org.testng.annotations.Test;
45 public class AlignViewControllerTest
48 @BeforeClass(alwaysRun = true)
49 public void setUpJvOptionPane()
51 JvOptionPane.setInteractiveMode(false);
52 JvOptionPane.setMockResponse(JvOptionPane.CANCEL_OPTION);
55 @Test(groups = "Functional")
56 public void testFindColumnsWithFeature()
58 SequenceI seq1 = new Sequence("seq1", "aMMMaaaaaaaaaaaaaaaa");
59 SequenceI seq2 = new Sequence("seq2", "aaaMMMMMMMaaaaaaaaaa");
60 SequenceI seq3 = new Sequence("seq3", "aaaaaaaaaaMMMMMaaaaa");
61 SequenceI seq4 = new Sequence("seq3", "aaaaaaaaaaaaaaaaaaaa");
64 * features start/end are base 1
66 seq1.addSequenceFeature(new SequenceFeature("Metal", "desc", 2, 4, 0f,
68 seq1.addSequenceFeature(new SequenceFeature("Helix", "desc", 1, 15, 0f,
70 seq2.addSequenceFeature(new SequenceFeature("Metal", "desc", 4, 10, 0f,
72 seq3.addSequenceFeature(new SequenceFeature("Metal", "desc", 11, 15,
76 * select the first three columns --> Metal in seq1 2-3
78 SequenceGroup sg = new SequenceGroup();
79 sg.setStartRes(0); // base 0
81 sg.addSequence(seq1, false);
82 sg.addSequence(seq2, false);
83 sg.addSequence(seq3, false);
84 sg.addSequence(seq4, false);
86 BitSet bs = new BitSet();
87 int seqCount = AlignViewController.findColumnsWithFeature("Metal", sg,
89 assertEquals(1, seqCount);
90 assertEquals(2, bs.cardinality());
91 assertTrue(bs.get(1));
92 assertTrue(bs.get(2));
95 * select the first four columns: Metal in seq1 2:4, seq2 4:4
99 seqCount = AlignViewController.findColumnsWithFeature("Metal", sg, bs);
100 assertEquals(2, seqCount);
101 assertEquals(3, bs.cardinality());
102 assertTrue(bs.get(1));
103 assertTrue(bs.get(2));
104 assertTrue(bs.get(3));
107 * select column 11: Metal in seq3 only
112 seqCount = AlignViewController.findColumnsWithFeature("Metal", sg, bs);
113 assertEquals(1, seqCount);
114 assertEquals(1, bs.cardinality());
115 assertTrue(bs.get(10));
118 * select columns 16-20: no Metal feature
123 seqCount = AlignViewController.findColumnsWithFeature("Metal", sg, bs);
124 assertEquals(0, seqCount);
125 assertEquals(0, bs.cardinality());
128 * look for a feature that isn't there
133 seqCount = AlignViewController.findColumnsWithFeature("Pfam", sg, bs);
134 assertEquals(0, seqCount);
135 assertEquals(0, bs.cardinality());
139 * shameless copy of test data from findFeature for testing mark columns from
142 @Test(groups = "Functional")
143 public void testSelectColumnsWithHighlight()
145 AlignFrame af = new FileLoader().LoadFileWaitTillLoaded(
146 "seq1 aMMMaaaaaaaaaaaaaaaa\n" + "seq2 aaaMMMMMMMaaaaaaaaaa\n"
147 + "seq3 aaaaaaaaaaMMMMMaaaaa\n"
148 + "seq4 aaaaaaaaaaaaaaaaaaaa\n", FormatAdapter.PASTE);
150 SearchResultsI sr = new SearchResults();
151 SequenceI[] sqs = af.getViewport().getAlignment().getSequencesArray();
152 SequenceI seq1 = sqs[0];
153 SequenceI seq2 = sqs[1];
154 SequenceI seq3 = sqs[2];
155 SequenceI seq4 = sqs[3];
158 * features start/end are base 1
160 sr.addResult(seq1, 2, 4);
161 sr.addResult(seq2, 4, 10);
162 sr.addResult(seq3, 11, 15);
165 * test Match/Find works first
167 Finder f = new Finder(af.getViewport().getAlignment(), null);
169 f.setCaseSensitive(true);
172 "Finder found different set of results to manually created SearchResults",
173 sr, f.getSearchResults());
176 * now check simple mark columns from find operation
178 af.getViewport().setSearchResults(sr);
179 AlignViewControllerI avc = af.avc;
181 avc.markHighlightedColumns(false, false, false);
182 assertTrue("Didn't select highlighted columns", Arrays.deepEquals(af
183 .getViewport().getColumnSelection().getSelectedRanges()
184 .toArray(), new int[][] { { 1, 14 } }));