Update L2SVMTest.java

Jaybit0 · Sep 2, 2024 · cb44f7a · cb44f7a
1 parent d85363e
commit cb44f7a
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/src/test/java/org/apache/sysds/test/applications/L2SVMTest.java b/src/test/java/org/apache/sysds/test/applications/L2SVMTest.java
@@ -67,7 +67,16 @@ public void setUp() {
 	}
 
 	@Test
-	public void testL2SVM()
+	public void testL2SVM1() {
+		testL2SVM(true);
+	}
+
+	@Test
+	public void testL2SVM2() {
+		testL2SVM(false);
+	}
+
+	private void testL2SVM(boolean ngrams)
 	{
 		System.out.println("------------ BEGIN " + TEST_NAME 
 			+ " TEST WITH {" + numRecords + ", " + numFeatures
@@ -83,9 +92,11 @@ public void testL2SVM()
 
 		List<String> proArgs = new ArrayList<>();
 		proArgs.add("-stats");
-		proArgs.add("-ngrams");
-		proArgs.add("3,2");
-		proArgs.add("10");
+		if (ngrams) {
+			proArgs.add("-ngrams");
+			proArgs.add("3,2");
+			proArgs.add("10");
+		}
 		proArgs.add("-nvargs");
 		proArgs.add("X=" + input("X"));
 		proArgs.add("Y=" + input("Y"));