Skip to content

Commit e657ed9

Browse files
committed
add tests for replicate of Matrix
1 parent 1267949 commit e657ed9

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

src/test/scala/testmatrix.scala

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -502,4 +502,19 @@ class TestMatrix extends munit.FunSuite {
502502
val expected = 1.0*4 + 2*5 + 3*6
503503
A.assertEquals(result, expected)
504504
}
505+
test("Replicate"){
506+
val Amat = Array(
507+
Array(2.0, 1.0, 3.0),
508+
Array(1.0, 2.0, 4.0),
509+
Array(2.0, 2.0, 4.0))
510+
511+
val b = Matrix.replicate(Amat)
512+
513+
val (row1, col1) = Matrix.size(Amat)
514+
val (row2, col2) = Matrix.size(b)
515+
A.assertEquals(row1, row2)
516+
A.assertEquals(col1, col2)
517+
518+
A.assert(Matrix.elementwise_equal(Amat, b, 1e-6))
519+
}
505520
}

0 commit comments

Comments
 (0)