Fix a bug in alternative generation

codelion · Jan 7, 2017 · 404eb88 · 404eb88
1 parent f8b0def
commit 404eb88
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/main/java/com/sourceclear/gramtest/GeneratorVisitor.java b/src/main/java/com/sourceclear/gramtest/GeneratorVisitor.java
@@ -112,8 +112,7 @@ public List<String> visitAlternative(bnfParser.AlternativeContext ctx) {
     List<List<String>> comStr = new LinkedList<>();
     for(bnfParser.ElementContext ec1 : ctx.element()) {
       List<String> slist = visitElement(ec1);
-      if(slist.isEmpty()) return new LinkedList<>();
-      else comStr.add(slist);
+      if(!slist.isEmpty()) comStr.add(slist);
     }
     List<String> emptyStr = new LinkedList<>();
     emptyStr.add("");

diff --git a/src/test/java/com/sourceclear/gramtest/MainTest.java b/src/test/java/com/sourceclear/gramtest/MainTest.java
@@ -76,7 +76,7 @@ public void testMainStrGram() throws IOException {
     GeneratorVisitor extractor = new GeneratorVisitor();
     extractor.visit(tree);
     List<String> generatedTests = extractor.getTests();
-    Assert.assertEquals(100,generatedTests.size());
+    Assert.assertEquals(100, generatedTests.size());
   }
 
   /**