JAL-1270 tweaks to AlignmentGenerator / client class

[jalview.git] / test / jalview / analysis / AlignmentGenerator.java

diff --git a/test/jalview/analysis/AlignmentGenerator.java b/test/jalview/analysis/AlignmentGenerator.java
index 0e3bbbb..3187fd9 100644 (file)
--- a/test/jalview/analysis/AlignmentGenerator.java
+++ b/test/jalview/analysis/AlignmentGenerator.java
@@ -53,7 +53,6 @@ import org.testng.annotations.BeforeClass;
  */
 public class AlignmentGenerator
 {
-
   @BeforeClass(alwaysRun = true)
   public void setUpJvOptionPane()
   {
@@ -87,13 +86,14 @@ public class AlignmentGenerator
       usage();
       return;
     }
-    BASES = args[0].toLowerCase().startsWith("n") ? NUCS : PEPS;
+    boolean nucleotide = args[0].toLowerCase().startsWith("n");
     int width = Integer.parseInt(args[1]);
     int height = Integer.parseInt(args[2]);
     long randomSeed = Long.valueOf(args[3]);
     int gapPercentage = Integer.valueOf(args[4]);
     int changePercentage = Integer.valueOf(args[5]);
-    AlignmentI al = new AlignmentGenerator().generate(width, height,
+    AlignmentI al = new AlignmentGenerator(nucleotide).generate(width,
+            height,
             randomSeed, gapPercentage, changePercentage);
 
     System.out.println("; " + height + " sequences of " + width
@@ -124,11 +124,11 @@ public class AlignmentGenerator
   }
 
   /**
-   * Default constructor
+   * Constructor that sets nucleotide or peptide symbol set
    */
-  public AlignmentGenerator()
+  public AlignmentGenerator(boolean nuc)
   {
-
+    BASES = nuc ? NUCS : PEPS;
   }
 
   /**