JAL-2388 last separation of hidden columns/col selection

[jalview.git] / test / jalview / datamodel / ColumnSelectionTest.java

diff --git a/test/jalview/datamodel/ColumnSelectionTest.java b/test/jalview/datamodel/ColumnSelectionTest.java
index 6debd87..ce2949f 100644 (file)
--- a/test/jalview/datamodel/ColumnSelectionTest.java
+++ b/test/jalview/datamodel/ColumnSelectionTest.java
@@ -59,6 +59,30 @@ public class ColumnSelectionTest
     assertEquals("[2, 5, 3]", sel.toString());
   }
 
+  @Test(groups = { "Functional" })
+  public void testSetElementsFrom()
+  {
+    ColumnSelection fromcs = new ColumnSelection();
+    ColumnSelection tocs = new ColumnSelection();
+    HiddenColumns hidden = new HiddenColumns();
+
+    fromcs.addElement(2);
+    fromcs.addElement(3);
+    fromcs.addElement(5);
+
+    tocs.setElementsFrom(fromcs, hidden);
+    assertTrue(tocs.equals(fromcs));
+
+    hidden.hideColumns(4, 6);
+    tocs.setElementsFrom(fromcs, hidden);
+
+    // expect cols 2 and 3 to be selected but not 5
+    ColumnSelection expectcs = new ColumnSelection();
+    expectcs.addElement(2);
+    expectcs.addElement(3);
+    assertTrue(tocs.equals(expectcs));
+  }
+
   /**
    * Test the remove method - in particular to verify that remove(int i) removes
    * the element whose value is i, _NOT_ the i'th element.
@@ -106,7 +130,7 @@ public class ColumnSelectionTest
     cs.addElement(5);
     cs.addElement(6);
     // hide column 5 (and adjacent):
-    cs.hideSelectedColumns(5, al);
+    cs.hideSelectedColumns(5, al.getHiddenColumns());
     // 4,5,6 now hidden:
     List<int[]> hidden = al.getHiddenColumns().getListOfCols();
     assertEquals(1, hidden.size());
@@ -120,7 +144,7 @@ public class ColumnSelectionTest
     cs.addElement(4);
     cs.addElement(5);
     cs.addElement(6);
-    cs.hideSelectedColumns(4, al);
+    cs.hideSelectedColumns(4, al.getHiddenColumns());
     hidden = al.getHiddenColumns().getListOfCols();
     assertEquals(1, hidden.size());
     assertEquals("[4, 6]", Arrays.toString(hidden.get(0)));
@@ -132,7 +156,7 @@ public class ColumnSelectionTest
     cs.addElement(4);
     cs.addElement(5);
     cs.addElement(6);
-    cs.hideSelectedColumns(6, al);
+    cs.hideSelectedColumns(6, al.getHiddenColumns());
     hidden = al.getHiddenColumns().getListOfCols();
     assertEquals(1, hidden.size());
     assertEquals("[4, 6]", Arrays.toString(hidden.get(0)));
@@ -143,7 +167,7 @@ public class ColumnSelectionTest
     cs = new ColumnSelection();
     cs.addElement(4);
     cs.addElement(6);
-    cs.hideSelectedColumns(5, al);
+    cs.hideSelectedColumns(5, al.getHiddenColumns());
     hidden = al.getHiddenColumns().getListOfCols();
     assertEquals(1, hidden.size());
     assertEquals("[4, 6]", Arrays.toString(hidden.get(0)));