test/jalview/schemes/UserColourSchemeTest.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.schemes;
  22
  23 import static org.testng.AssertJUnit.assertEquals;
  24
  25 import jalview.gui.JvOptionPane;
  26
  27 import java.awt.Color;
  28
  29 import org.testng.annotations.BeforeClass;
  30 import org.testng.annotations.Test;
  31
  32 public class UserColourSchemeTest
  33 {
  34
  35   @BeforeClass(alwaysRun = true)
  36   public void setUpJvOptionPane()
  37   {
  38     JvOptionPane.setInteractiveMode(false);
  39     JvOptionPane.setMockResponse(JvOptionPane.CANCEL_OPTION);
  40   }
  41
  42   @Test(groups = "Functional")
  43   public void testParseAppletParameter()
  44   {
  45     UserColourScheme cs = new UserColourScheme("white");
  46     cs.parseAppletParameter("D,E=red; K,R,H=0022FF; c=10 , 20,30");
  47     assertEquals(Color.RED, cs.findColour('D'));
  48     assertEquals(Color.RED, cs.findColour('d'));
  49     assertEquals(Color.RED, cs.findColour('E'));
  50     assertEquals(Color.RED, cs.findColour('e'));
  51     Color c1 = new Color(0x0022ff);
  52     assertEquals(c1, cs.findColour('K'));
  53     assertEquals(c1, cs.findColour('R'));
  54     assertEquals(c1, cs.findColour('h'));
  55     Color c2 = new Color(10, 20, 30);
  56     assertEquals(c2, cs.findColour('c'));
  57     assertEquals(Color.WHITE, cs.findColour('G'));
  58     assertEquals(Color.WHITE, cs.findColour('-'));
  59     assertEquals(Color.WHITE, cs.findColour('.'));
  60     assertEquals(Color.WHITE, cs.findColour(' '));
  61
  62     cs = new UserColourScheme("white");
  63     cs.parseAppletParameter(
  64             "D,E=red; K,R,H=0022FF; c=10 , 20,30;t=orange;lowercase=blue;s=pink");
  65     assertEquals(Color.RED, cs.findColour('D'));
  66     assertEquals(Color.blue, cs.findColour('d'));
  67     assertEquals(Color.RED, cs.findColour('E'));
  68     assertEquals(Color.blue, cs.findColour('e'));
  69     assertEquals(c1, cs.findColour('K'));
  70     assertEquals(c1, cs.findColour('R'));
  71     assertEquals(Color.blue, cs.findColour('h'));
  72     assertEquals(c2, cs.findColour('c'));
  73     // 'lowercase' sets all lower-case not already set to the given colour
  74     assertEquals(Color.orange, cs.findColour('t'));
  75     assertEquals(Color.blue, cs.findColour('k'));
  76     assertEquals(Color.blue, cs.findColour('a'));
  77     assertEquals(Color.pink, cs.findColour('s'));
  78   }
  79
  80   @Test(groups = "Functional")
  81   public void testToAppletParameter()
  82   {
  83     UserColourScheme cs = new UserColourScheme(
  84             "E,D=red; K,R,H=0022FF; c=10 , 20,30");
  85     String param = cs.toAppletParameter();
  86     assertEquals("D,E=ff0000;H,K,R=0022ff;c=0a141e", param);
  87   }
  88
  89   /**
  90    * Test for user colour scheme constructed with a colour per residue,
  91    * including gap. Note this can currently be done from the User Defined
  92    * Colours dialog, but not by parsing a colours parameter, as
  93    * parseAppletParameter only recognises amino acid codes.
  94    */
  95   @Test(groups = "Functional")
  96   public void testConstructor_coloursArray()
  97   {
  98     Color g = Color.green;
  99     Color y = Color.yellow;
 100     Color b = Color.blue;
 101     Color r = Color.red;
 102     // colours for ARNDCQEGHILKMFPSTWYVBZ and gap
 103     Color[] colours = new Color[] { g, y, b, r, g, y, r, b, g, y, r, b, g,
 104         y, r, b, g, y, r, b, g, y, r, g };
 105     UserColourScheme cs = new UserColourScheme(colours);
 106
 107     assertEquals(g, cs.findColour('A'));
 108     assertEquals(b, cs.findColour('n'));
 109     assertEquals(g, cs.findColour('-'));
 110     assertEquals(g, cs.findColour('.'));
 111     assertEquals(g, cs.findColour(' '));
 112   }
 113 }