feat(test/component/estim): add unit tests for row wise sparsity estimator with element-wise and single operations

Author: ywcb00

src/test/java/org/apache/sysds/test/component/estim/OpElemWChainTest.java

@@ -25,6 +25,7 @@
 import org.apache.sysds.hops.estim.EstimatorBitsetMM;
 import org.apache.sysds.hops.estim.EstimatorLayeredGraph;
 import org.apache.sysds.hops.estim.EstimatorMatrixHistogram;
+import org.apache.sysds.hops.estim.EstimatorRowWise;
 import org.apache.sysds.hops.estim.EstimatorDensityMap;
 import org.apache.sysds.hops.estim.MMNode;
 import org.apache.sysds.hops.estim.SparsityEstimator;
@@ -118,8 +119,18 @@ public void testLGCasemult() {
 	public void testLGCaseplus() {
 		runSparsityEstimateTest(new EstimatorLayeredGraph(), m, n, sparsity, plus);
 	}
-	
-	
+
+	// Row Wise Sparsity Estimator
+	@Test
+	public void testRowWiseCaseMult() {
+		runSparsityEstimateTest(new EstimatorRowWise(), m, n, sparsity, mult);
+	}
+
+	@Test
+	public void testRowWiseCasePlus() {
+		runSparsityEstimateTest(new EstimatorRowWise(), m, n, sparsity, plus);
+	}
+
 	private static void runSparsityEstimateTest(SparsityEstimator estim, int m, int n, double[] sp, OpCode op) {
 		MatrixBlock m1 = MatrixBlock.randOperations(m, n, sp[0], 1, 1, "uniform", 3);
 		MatrixBlock m2 = MatrixBlock.randOperations(m, n, sp[1], 1, 1, "uniform", 5);

src/test/java/org/apache/sysds/test/component/estim/OpElemWTest.java

@@ -25,6 +25,7 @@
 import org.apache.sysds.hops.estim.EstimatorBitsetMM;
 import org.apache.sysds.hops.estim.EstimatorDensityMap;
 import org.apache.sysds.hops.estim.EstimatorMatrixHistogram;
+import org.apache.sysds.hops.estim.EstimatorRowWise;
 import org.apache.sysds.hops.estim.EstimatorLayeredGraph;
 import org.apache.sysds.hops.estim.EstimatorSample;
 import org.apache.sysds.hops.estim.SparsityEstimator;
@@ -128,7 +129,18 @@ public void testSampleMult() {
 	public void testSamplePlus() {
 		runSparsityEstimateTest(new EstimatorSample(), m, n, sparsity, plus);
 	}
-	
+
+	// Row Wise Sparsity Estimator
+	@Test
+	public void testRowWiseMult() {
+		runSparsityEstimateTest(new EstimatorRowWise(), m, n, sparsity, mult);
+	}
+
+	@Test
+	public void testRowWisePlus() {
+		runSparsityEstimateTest(new EstimatorRowWise(), m, n, sparsity, plus);
+	}
+
 	private static void runSparsityEstimateTest(SparsityEstimator estim, int m, int n, double[] sp, OpCode op) {
 		MatrixBlock m1 = MatrixBlock.randOperations(m, n, sp[0], 1, 1, "uniform", 3);
 		MatrixBlock m2 = MatrixBlock.randOperations(m, n, sp[1], 1, 1, "uniform", 7);

src/test/java/org/apache/sysds/test/component/estim/OpSingleTest.java

@@ -26,6 +26,7 @@
 import org.apache.sysds.hops.estim.EstimatorBasicWorst;
 import org.apache.sysds.hops.estim.EstimatorBitsetMM;
 import org.apache.sysds.hops.estim.EstimatorLayeredGraph;
+import org.apache.sysds.hops.estim.EstimatorRowWise;
 import org.apache.sysds.hops.estim.SparsityEstimator;
 import org.apache.sysds.hops.estim.SparsityEstimator.OpCode;
 import org.apache.sysds.runtime.matrix.data.MatrixBlock;
@@ -40,7 +41,7 @@ public class OpSingleTest extends AutomatedTestBase
 	private final static int m = 600;
 	private final static int k = 300;
 	private final static double sparsity = 0.2;
-//	private final static OpCode eqzero = OpCode.EQZERO;
+	// private final static OpCode eqzero = OpCode.EQZERO;
 	private final static OpCode diag = OpCode.DIAG;
 	private final static OpCode neqzero = OpCode.NEQZERO;
 	private final static OpCode trans = OpCode.TRANS;
@@ -237,7 +238,33 @@ public void testLGCasetrans() {
 //	public void testSampleCasereshape() {
 //		runSparsityEstimateTest(new EstimatorSample(), m, k, sparsity, reshape);
 //	}
-	
+
+	// Row Wise Sparsity Estimator
+	// @Test
+	// public void testRowWiseEqzero() {
+	// 	runSparsityEstimateTest(new EstimatorRowWise(), m, k, sparsity, eqzero);
+	// }
+
+	// @Test
+	// public void testRowWiseDiag() {
+	// 	runSparsityEstimateTest(new EstimatorRowWise(), m, m, sparsity, diag);
+	// }
+
+	@Test
+	public void testRowWiseNeqzero() {
+		runSparsityEstimateTest(new EstimatorRowWise(), m, k, sparsity, neqzero);
+	}
+
+	@Test
+	public void testRowWiseTrans() {
+		runSparsityEstimateTest(new EstimatorRowWise(), m, k, sparsity, trans);
+	}
+
+	@Test
+	public void testRowWiseReshape() {
+		runSparsityEstimateTest(new EstimatorRowWise(), m, k, sparsity, reshape);
+	}
+
 	private static void runSparsityEstimateTest(SparsityEstimator estim, int m, int k, double sp, OpCode op) {
 		MatrixBlock m1 = MatrixBlock.randOperations(m, k, sp, 1, 1, "uniform", 3);
 		MatrixBlock m2 = new MatrixBlock();
@@ -252,13 +279,7 @@ private static void runSparsityEstimateTest(SparsityEstimator estim, int m, int
 				est = estim.estim(m1, op);
 				break;
 			case NEQZERO:
-				m2 = m1;
-				est = estim.estim(m1, op);
-				break;
 			case TRANS:
-				m2 = m1;
-				est = estim.estim(m1, op);
-				break;
 			case RESHAPE:
 				m2 = m1;
 				est = estim.estim(m1, op);