public void testRowsIndexed() {
SparseMatrixLil A = new SparseMatrixLil("1 5 3; 2 8 5; 1 9 4; 2 5 3");
DenseMatrix indexes = new DenseMatrix("2; 1; 1");
SparseMatrixLil B = A.rows(indexes);
System.out.println(B.toDense());
assertTrue(B.equals(new SparseMatrixLil("1 9 4; 2 8 5; 2 8 5")));
}
public void testSparseMultiply() {
SparseMatrixLil A = sprand(3, 5);
System.out.println(A);
SparseMatrixLil A2 = A.mmul(speye(5));
System.out.println(A2);
assertTrue(A.equals(A2));
}
public void testConstructorByStringSparse() {
SparseMatrixLil foo = new SparseMatrixLil("1 0; 5 7; 0 5");
assertEquals(3, foo.rows);
assertEquals(2, foo.cols);
assertTrue(foo.equals(new DenseMatrix(new double[][] {{1, 0}, {5, 7}, {0, 5}})));
assertEquals(5.0, foo.toDense().get(1, 0));
assertEquals(7.0, foo.toDense().get(1, 1));
assertEquals(4, foo.size);
}
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));
}
public void testSparseDenseMmul() {
int K = 60;
int N = thousandconstant;
SparseMatrixLil A = SparseMatrixLil.rand(K, N);
DenseMatrix B = DenseMatrix.rand(N, K);
Timer timer = new Timer();
DenseMatrix C = A.mmul(B);
timer.printTimeCheckMilliseconds();
DenseMatrix C2 = A.toDense().mmul(B);
assertTrue(C.equals(C2));
}
public void testTranspose() {
DenseMatrix A = rand(5, 8);
DenseMatrix At = A.t();
assertEquals(5, At.cols);
assertEquals(8, At.rows);
assertTrue(A.equals(At.t()));
SparseMatrixLil B = A.toSparseLil();
SparseMatrixLil Bt = B.t();
assertEquals(5, Bt.cols);
assertEquals(8, Bt.rows);
assertTrue(At.equals(Bt));
assertTrue(A.equals(Bt.t()));
}
public void testOneSparse() {
SparseMatrixLil A = SparseMatrixLil.rand(3, 3).t();
System.out.println(A);
SparseMatrixLil B = SparseMatrixLil.rand(3, 3);
System.out.println(B);
SparseMatrixLil C = A.mmul(B);
System.out.println(C);
DenseMatrix CDense = A.toDense().mmul(B.toDense());
System.out.println(CDense);
assertTrue(C.equals(CDense));
}
public void testToCCS() {
SparseMatrixLil A = sprand(10, 10);
SparseMatrixLil B = A.toCCS().toLil();
assertTrue(A.equals(B));
A = sprand(thousandconstant, thousandconstant);
B = A.toCCS().toLil();
assertTrue(A.equals(B));
A = A.t();
B = A.toCCS().toLil();
assertTrue(A.equals(B));
}
public void testSortBigMatrixInplace() { // you will need to add option -Xmx1400m to run this
SparseMatrixLil B;
SparseMatrixLil A = sprand(thousandconstant, largematrixsize);
System.out.println("size: " + A.getSize());
System.out.println("created A");
B = A.add(0); // make copy
System.out.println("created B");
// System.out.println(B);
B.sortInplace();
for (int i = 0; i < B.size; i++) {
if (i > 0) {
assertTrue(B.colIdx[i] >= B.colIdx[i - 1]);
if (B.colIdx[i] == B.colIdx[i - 1]) {
assertTrue(B.rowIdx[i] >= B.rowIdx[i - 1]);
}
}
assertEquals(A.rowIdx[i], B.rowIdx[i]);
assertEquals(A.colIdx[i], B.colIdx[i]);
assertEquals(A.values[i], B.values[i]);
}
System.out.println("sorted B");
// assertTrue( A.equals(B));
// System.out.println("asserted equals");
// System.out.println(B);
B = null;
SparseMatrixLil Bt = A.t(); // t() is also a copy
System.out.println("transposed A");
// System.out.println(Bt);
Timer timer = new Timer();
Bt.sortInplace();
timer.printTimeCheckMilliseconds();
System.out.println("sorted Bt");
for (int i = 0; i < Bt.size; i++) {
if (i > 0) {
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));
System.out.println("checed equal to At");
}
public void xtestSortViaNative() {
SparseMatrixLil B = spzeros(2, 2);
System.out.println("xtestSortViaNative");
int R = thousandconstant;
int C = largematrixsize;
SparseMatrixLil A = sprand(R, C);
System.out.println("created A");
B = A.add(0); // make copy
System.out.println("created B");
// System.out.println(B);
// B.sort();
B = B.mmul(speye(C));
for (int i = 0; i < B.size; i++) {
if (i > 0) {
assertTrue(B.colIdx[i] >= B.colIdx[i - 1]);
if (B.colIdx[i] == B.colIdx[i - 1]) {
assertTrue(B.rowIdx[i] >= B.rowIdx[i - 1]);
}
}
assertEquals(A.rowIdx[i], B.rowIdx[i]);
assertEquals(A.colIdx[i], B.colIdx[i]);
assertEquals(A.values[i], B.values[i]);
}
System.out.println("sorted B");
// assertTrue( A.equals(B));
// System.out.println("asserted equals");
// System.out.println(B);
B = null;
SparseMatrixLil Bt = A.t(); // t() is also a copy
System.out.println("transposed A");
// System.out.println(Bt);
// Bt.sort();
Bt = Bt.mmul(speye(R));
System.out.println("sorted Bt");
for (int i = 0; i < Bt.size; i++) {
if (i > 0) {
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));
System.out.println("checed equal to At");
}
public static final DenseMatrix meanOverRows(SparseMatrixLil mat) {
return mat.sumOverRows().div(mat.rows);
}
public void testThreeSparse() {
int K = 60;
int N = thousandconstant;
SparseMatrixLil A = SparseMatrixLil.rand(N, K);
SparseMatrixLil B = SparseMatrixLil.rand(K, N);
// System.out.println(A.toDense());
Timer timer = new Timer();
SparseMatrixLil C = B.mmul(A);
timer.printTimeCheckMilliseconds();
SparseMatrixLil C1 = B.mmul(A);
timer.printTimeCheckMilliseconds();
SparseMatrixLil C2 = B.mmul(A);
timer.printTimeCheckMilliseconds();
DenseMatrix CDense = B.toDense().mmul(A.toDense());
timer.printTimeCheckMilliseconds("for dense");
assertTrue(C.equals(CDense));
K = 400;
N = thousandconstant;
A = SparseMatrixLil.rand(N, K);
B = SparseMatrixLil.rand(K, N);
// System.out.println(A.toDense());
timer = new Timer();
C = B.mmul(A);
timer.printTimeCheckMilliseconds();
C1 = B.mmul(A);
timer.printTimeCheckMilliseconds();
C2 = B.mmul(A);
timer.printTimeCheckMilliseconds();
CDense = B.toDense().mmul(A.toDense());
timer.printTimeCheckMilliseconds("for dense");
assertTrue(C.equals(CDense));
}
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))));
}
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);
}
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());
}
public void testMin() {
SparseMatrixLil A = speye(5);
DenseMatrix B = A.minOverCols();
System.out.println(B);
assertEquals(5.0, B.sumOverRows().s());
}
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));
}
public static final DenseMatrix varOverRows(SparseMatrixLil mat) {
return mat.pow(2).sumOverRows().sub(mat.sumOverRows().pow(2).div(mat.rows)).div(mat.rows - 1);
}