From 404eb8834653b58dde9c0fcb4cad840bf9743c81 Mon Sep 17 00:00:00 2001
From: Asankhaya Sharma <asankhaya@yahoo.com>
Date: Sat, 7 Jan 2017 08:57:09 +0800
Subject: [PATCH] Fix a bug in alternative generation

---
 src/main/java/com/sourceclear/gramtest/GeneratorVisitor.java | 3 +--
 src/test/java/com/sourceclear/gramtest/MainTest.java         | 2 +-
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/sourceclear/gramtest/GeneratorVisitor.java b/src/main/java/com/sourceclear/gramtest/GeneratorVisitor.java
index c424e23..8632b6f 100644
--- 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
index 21a23ff..20c1875 100644
--- 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());
   }
 
   /**