X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=test%2Fjalview%2Fext%2Frbvi%2Fchimera%2FChimeraCommandsTest.java;h=d3511712221152cf361184538647d21616304e01;hb=799c26111d6936a2e70cb5f1fd7d7312311e6db9;hp=fb442e31e8f8bf02b4442335cffbd928bb53d425;hpb=b55fd078dae331c368cd90165459a6b0efe6305d;p=jalview.git diff --git a/test/jalview/ext/rbvi/chimera/ChimeraCommandsTest.java b/test/jalview/ext/rbvi/chimera/ChimeraCommandsTest.java index fb442e3..d351171 100644 --- a/test/jalview/ext/rbvi/chimera/ChimeraCommandsTest.java +++ b/test/jalview/ext/rbvi/chimera/ChimeraCommandsTest.java @@ -23,7 +23,18 @@ package jalview.ext.rbvi.chimera; import static org.testng.Assert.assertEquals; import static org.testng.Assert.assertTrue; +import jalview.datamodel.Alignment; +import jalview.datamodel.AlignmentI; +import jalview.datamodel.ColumnSelection; +import jalview.datamodel.Sequence; +import jalview.datamodel.SequenceI; +import jalview.gui.AlignFrame; import jalview.gui.JvOptionPane; +import jalview.gui.SequenceRenderer; +import jalview.schemes.JalviewColourScheme; +import jalview.structure.StructureMapping; +import jalview.structure.StructureMappingcommandSet; +import jalview.structure.StructureSelectionManager; import java.awt.Color; import java.util.HashMap; @@ -90,7 +101,7 @@ public class ChimeraCommandsTest * feature name gets a jv_ namespace prefix * feature value is quoted in case it contains spaces */ - assertEquals(commands.get(0), "setattr r jv_chain \"X\" #0:8-20.A"); + assertEquals(commands.get(0), "setattr r jv_chain 'X' #0:8-20.A"); // add same feature value, overlapping range ChimeraCommands.addColourRange(featureValues, "X", 0, 3, 9, "A"); @@ -98,7 +109,7 @@ public class ChimeraCommandsTest ChimeraCommands.addColourRange(featureValues, "X", 0, 21, 25, "A"); commands = ChimeraCommands.buildSetAttributeCommands(featuresMap); assertEquals(1, commands.size()); - assertEquals(commands.get(0), "setattr r jv_chain \"X\" #0:3-25.A"); + assertEquals(commands.get(0), "setattr r jv_chain 'X' #0:3-25.A"); // same feature value and model, different chain ChimeraCommands.addColourRange(featureValues, "X", 0, 21, 25, "B"); @@ -107,7 +118,7 @@ public class ChimeraCommandsTest commands = ChimeraCommands.buildSetAttributeCommands(featuresMap); assertEquals(1, commands.size()); assertEquals(commands.get(0), - "setattr r jv_chain \"X\" #0:3-25.A,21-25.B|#1:26-30.A"); + "setattr r jv_chain 'X' #0:3-25.A,21-25.B|#1:26-30.A"); // same feature, different value ChimeraCommands.addColourRange(featureValues, "Y", 0, 40, 50, "A"); @@ -116,18 +127,20 @@ public class ChimeraCommandsTest // commands are ordered by feature type but not by value // so use contains to test for the expected command: assertTrue(commands - .contains("setattr r jv_chain \"X\" #0:3-25.A,21-25.B|#1:26-30.A")); - assertTrue(commands.contains("setattr r jv_chain \"Y\" #0:40-50.A")); + .contains("setattr r jv_chain 'X' #0:3-25.A,21-25.B|#1:26-30.A")); + assertTrue(commands.contains("setattr r jv_chain 'Y' #0:40-50.A")); featuresMap.clear(); featureValues.clear(); featuresMap.put("side-chain binding!", featureValues); - ChimeraCommands.addColourRange(featureValues, "metal ion!", 0, 7, 15, + ChimeraCommands.addColourRange(featureValues, + "metal 'ion!", 0, 7, 15, "A"); - // feature names are sanitised to change space or hyphen to underscore + // feature names are sanitised to change non-alphanumeric to underscore + // feature values are sanitised to encode single quote characters commands = ChimeraCommands.buildSetAttributeCommands(featuresMap); assertTrue(commands - .contains("setattr r jv_side_chain_binding_ \"metal ion!\" #0:7-15.A")); + .contains("setattr r jv_side_chain_binding_ 'metal 'ion!' #0:7-15.A")); } /** @@ -149,4 +162,60 @@ public class ChimeraCommandsTest assertEquals(ChimeraCommands.makeAttributeName("helixColor"), "jv_helixColor_"); } + + @Test(groups = { "Functional" }) + public void testGetColourBySequenceCommands_hiddenColumns() + { + /* + * load these sequences, coloured by Strand propensity, + * with columns 2-4 hidden + */ + SequenceI seq1 = new Sequence("seq1", "MHRSQSSSGG"); + SequenceI seq2 = new Sequence("seq2", "MVRSNGGSSS"); + AlignmentI al = new Alignment(new SequenceI[] { seq1, seq2 }); + AlignFrame af = new AlignFrame(al, 800, 500); + af.changeColour_actionPerformed(JalviewColourScheme.Strand.toString()); + ColumnSelection cs = new ColumnSelection(); + cs.addElement(2); + cs.addElement(3); + cs.addElement(4); + af.getViewport().setColumnSelection(cs); + af.hideSelColumns_actionPerformed(null); + SequenceRenderer sr = new SequenceRenderer(af.getViewport()); + SequenceI[][] seqs = new SequenceI[][] { { seq1 }, { seq2 } }; + String[] files = new String[] { "seq1.pdb", "seq2.pdb" }; + StructureSelectionManager ssm = new StructureSelectionManager(); + + /* + * map residues 1-10 to residues 21-30 (atoms 105-150) in structures + */ + HashMap map = new HashMap(); + for (int pos = 1; pos <= seq1.getLength(); pos++) + { + map.put(pos, new int[] { 20 + pos, 5 * (20 + pos) }); + } + StructureMapping sm1 = new StructureMapping(seq1, "seq1.pdb", "pdb1", + "A", map, null); + ssm.addStructureMapping(sm1); + StructureMapping sm2 = new StructureMapping(seq2, "seq2.pdb", "pdb2", + "B", map, null); + ssm.addStructureMapping(sm2); + + StructureMappingcommandSet[] commands = ChimeraCommands + .getColourBySequenceCommand(ssm, files, seqs, sr, null, + af.getViewport()); + assertEquals(1, commands.length); + assertEquals(1, commands[0].commands.length); + String theCommand = commands[0].commands[0]; + // M colour is #82827d (see strand.html help page) + assertTrue(theCommand.contains("color #82827d #0:21.A|#1:21.B")); + // H colour is #60609f + assertTrue(theCommand.contains("color #60609f #0:22.A")); + // V colour is ##ffff00 + assertTrue(theCommand.contains("color #ffff00 #1:22.B")); + // hidden columns are Gray (128, 128, 128) + assertTrue(theCommand.contains("color #808080 #0:23-25.A|#1:23-25.B")); + // S and G are both coloured #4949b6 + assertTrue(theCommand.contains("color #4949b6 #0:26-30.A|#1:26-30.B")); + } }