Exemplo n.º 1
0
 public void testCCsGetNonZeros() {
   SparseMatrixLil A = spzeros(3, 3);
   A.append(1, 0, 5);
   A.append(0, 1, 7);
   A.append(0, 0, 11);
   SparseMatrixCCS B = A.toCCS();
   assertEquals(3, B.nonZeros());
   assertEquals(2, B.nonZeros(0));
   assertEquals(1, B.nonZeros(1));
   assertEquals(0, B.nonZeros(2));
 }
Exemplo n.º 2
0
 public void testCreateSparse() {
   DenseMatrix A = rand(5, 5);
   SparseMatrixLil B = A.toSparseLil();
   assertTrue(B.equals(B.mmul(eye(5))));
   assertTrue(B.equals(B.t().mmul(eye(5)).t()));
   B = spzeros(2, 2);
   B.append(1, 0, 5);
   B.append(0, 1, 7);
   B.append(0, 0, 11);
   System.out.println(B.toDense());
   System.out.println(B.mmul(eye(2)));
   assertTrue(B.equals(B.mmul(eye(2))));
 }
Exemplo n.º 3
0
 public void testSum() {
   DenseMatrix A = new DenseMatrix(new double[][] {{1, 9}, {7, 3}});
   assertTrue(new DenseMatrix(new double[][] {{8, 12}}).equals(A.sumOverRows()));
   assertTrue(new DenseMatrix(new double[][] {{10}, {10}}).equals(A.sumOverCols()));
   SparseMatrixLil B = A.toSparseLil();
   assertTrue(new DenseMatrix(new double[][] {{8, 12}}).toSparseLil().equals(B.sumOverRows()));
   assertTrue(new DenseMatrix(new double[][] {{10}, {10}}).toSparseLil().equals(B.sumOverCols()));
   assertTrue(B.sumOverRows().equals(A.sumOverRows()));
   assertTrue(B.sumOverCols().equals(A.sumOverCols()));
   assertEquals(B.sum().sum().s(), A.sum().sum().s());
   SparseMatrixLil C = spzeros(4, 5);
   C.append(1, 2, 5);
   C.append(3, 1, 3);
   C.append(3, 2, 7);
   C.append(1, 4, 11);
   System.out.println(C.sumOverRows());
   assertTrue(new DenseMatrix(new double[][] {{0, 3, 12, 0, 11}}).equals(C.sumOverRows()));
   System.out.println(C.sumOverCols());
   assertTrue(new DenseMatrix(new double[][] {{0}, {16}, {0}, {10}}).equals(C.sumOverCols()));
   assertEquals(26.0, C.sumOverCols().sumOverRows().s());
   assertEquals(26.0, C.sumOverRows().sumOverCols().s());
 }
Exemplo n.º 4
0
  public void testSort() {
    SparseMatrixLil B = spzeros(2, 2);
    B.append(1, 0, 5);
    B.append(0, 1, 7);
    B.append(0, 0, 11);
    System.out.println(B.toDense());
    System.out.println(B);
    B.sort();
    System.out.println(B.toDense());
    System.out.println(B);
    assertEquals(11.0, B.values[0]);
    assertEquals(5.0, B.values[1]);
    assertEquals(7.0, B.values[2]);

    SparseMatrixLil A = sprand(5, 8);
    B = A.add(0); // make copy
    System.out.println(B);
    B.sort();
    for (int i = 1; i < B.size; i++) {
      assertTrue(B.colIdx[i] >= B.colIdx[i - 1]);
      if (B.colIdx[i] == B.colIdx[i - 1]) {
        assertTrue(B.rowIdx[i] >= B.rowIdx[i - 1]);
      }
    }
    assertTrue(A.equals(B));
    System.out.println(B);
    SparseMatrixLil Bt = B.t();
    System.out.println(Bt);
    Bt.sort();
    for (int i = 1; i < Bt.size; i++) {
      assertTrue(Bt.colIdx[i] >= Bt.colIdx[i - 1]);
      if (Bt.colIdx[i] == Bt.colIdx[i - 1]) {
        assertTrue(Bt.rowIdx[i] >= Bt.rowIdx[i - 1]);
      }
    }
    System.out.println(Bt);
    assertTrue(A.toDense().t().equals(Bt));
  }
Exemplo n.º 5
0
  public void testSliceSparse() {
    SparseMatrixLil A = sprand(5, 8);
    SparseMatrixLil B = sprand(5, 12);
    SparseMatrixLil C = sprand(4, 8);
    SparseMatrixLil AB = A.concatRight(B);
    assertTrue(A.equals(AB.cols(0, 8)));
    assertTrue(B.equals(AB.cols(8, 20)));
    SparseMatrixLil AC = A.concatDown(C);
    assertTrue(A.equals(AC.rows(0, 5)));
    assertTrue(C.equals(AC.rows(5, 9)));

    A = spzeros(1, 5);
    A.append(0, 2, 3);
    assertEquals(A.row(0).cols, 5);
  }