/*
* Copyright (c) 2009-2014, Peter Abeles. All Rights Reserved.
*
* This file is part of Efficient Java Matrix Library (EJML).
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.ejml.ops;
import org.ejml.data.*;
/**
* Contains various functions related to unit testing matrix operations.
*
* @author Peter Abeles
*/
public class EjmlUnitTests {
/**
* Checks to see if every element in A is countable. A doesn't have any element with
* a value of NaN or infinite.
*
* @param A Matrix
*/
public static void assertCountable( RealMatrix64F A ) {
for( int i = 0; i < A.getNumRows(); i++ ){
for( int j = 0; j < A.getNumCols(); j++ ) {
assertTrue( !Double.isNaN(A.get(i,j)) , "NaN found at "+i+" "+j );
assertTrue( !Double.isInfinite(A.get(i,j)) , "Infinite found at "+i+" "+j );
}
}
}
/**
*
* Checks to see if A and B have the same shape.
*
*
* @param A Matrix
* @param B Matrix
*/
public static void assertShape( Matrix A , Matrix B ) {
assertTrue( A.getNumRows() == B.getNumRows() , "Number of rows do not match");
assertTrue( A.getNumCols() == B.getNumCols() , "Number of columns do not match");
}
/**
*
* Checks to see if the matrix has the specified number of rows and columns.
*
*
* @param A Matrix
* @param numRows expected number of rows in the matrix
* @param numCols expected number of columns in the matrix
*/
public static void assertShape( RealMatrix64F A , int numRows , int numCols ) {
assertTrue( A.getNumRows() == numRows , "Unexpected number of rows.");
assertTrue( A.getNumCols() == numCols , "Unexpected number of columns.");
}
/**
*
* Checks to see if each element in the matrix is within tolerance of each other:
*
*
*
* The two matrices are identical with in tolerance if:
* |aij - bij| ≤ tol
*
*
*
* In addition if an element is NaN or infinite in one matrix it must be the same in the other.
*
*
* @param A Matrix A
* @param B Matrix B
* @param tol Tolerance
*/
public static void assertEqualsUncountable( RealMatrix64F A , RealMatrix64F B , double tol ) {
assertShape(A, B);
for (int i = 0; i < A.getNumRows(); i++) {
for (int j = 0; j < A.getNumCols(); j++) {
double valA = A.get(i, j);
double valB = B.get(i, j);
if (Double.isNaN(valA)) {
assertTrue(Double.isNaN(valB), "At (" + i + "," + j + ") A = " + valA + " B = " + valB);
} else if (Double.isInfinite(valA)) {
assertTrue(Double.isInfinite(valB), "At (" + i + "," + j + ") A = " + valA + " B = " + valB);
} else {
double diff = Math.abs(valA - valB);
assertTrue(diff <= tol, "At (" + i + "," + j + ") A = " + valA + " B = " + valB);
}
}
}
}
/**
*
* Checks to see if each element in the matrices are within tolerance of each other and countable:
*
*
*
* The two matrices are identical with in tolerance if:
* |aij - bij| ≤ tol
*
*
*
* The test will fail if any element in either matrix is NaN or infinite.
*
*
* @param A Matrix A
* @param B Matrix B
* @param tol Tolerance
*/
public static void assertEquals( RealMatrix64F A , RealMatrix64F B , double tol ) {
assertShape(A,B);
for( int i = 0; i < A.getNumRows(); i++ ){
for( int j = 0; j < A.getNumCols(); j++ ) {
double valA = A.get(i,j);
double valB = B.get(i,j);
assertTrue(!Double.isNaN(valA) && !Double.isNaN(valB) ,"At ("+i+","+j+") A = "+valA+" B = "+valB);
assertTrue(!Double.isInfinite(valA) && !Double.isInfinite(valB) ,"At ("+i+","+j+") A = "+valA+" B = "+valB);
assertTrue(Math.abs( valA-valB) <= tol,"At ("+i+","+j+") A = "+valA+" B = "+valB);
}
}
}
public static void assertEquals( RealMatrix32F A , RealMatrix32F B , float tol ) {
assertShape(A,B);
for( int i = 0; i < A.getNumRows(); i++ ){
for( int j = 0; j < A.getNumCols(); j++ ) {
float valA = A.get(i,j);
float valB = B.get(i,j);
assertTrue(!Float.isNaN(valA) && !Float.isNaN(valB) ,"At ("+i+","+j+") A = "+valA+" B = "+valB);
assertTrue(!Float.isInfinite(valA) && !Float.isInfinite(valB) ,"At ("+i+","+j+") A = "+valA+" B = "+valB);
assertTrue(Math.abs( valA-valB) <= tol,"At ("+i+","+j+") A = "+valA+" B = "+valB);
}
}
}
public static void assertEquals( Complex64F a , Complex64F b , double tol ) {
assertTrue(!Double.isNaN(a.real) && !Double.isNaN(b.real) ,"real a = "+a.real+" b = "+b.real);
assertTrue(!Double.isInfinite(a.real) && !Double.isInfinite(b.real) ,"real a = "+a.real+" b = "+b.real);
assertTrue(Math.abs( a.real-b.real) <= tol,"real a = "+a.real+" b = "+b.real);
assertTrue(!Double.isNaN(a.imaginary) && !Double.isNaN(b.imaginary) ,"imaginary a = "+a.imaginary+" b = "+b.imaginary);
assertTrue(!Double.isInfinite(a.imaginary) && !Double.isInfinite(b.imaginary) ,"imaginary a = "+a.imaginary+" b = "+b.imaginary);
assertTrue(Math.abs( a.imaginary-b.imaginary) <= tol,"imaginary a = "+a.imaginary+" b = "+b.imaginary);
}
public static void assertEquals( ComplexMatrix64F A , ComplexMatrix64F B , double tol ) {
assertShape(A,B);
Complex64F a = new Complex64F();
Complex64F b = new Complex64F();
for( int i = 0; i < A.getNumRows(); i++ ){
for( int j = 0; j < A.getNumCols(); j++ ) {
A.get(i, j, a);
B.get(i, j, b);
assertTrue(!Double.isNaN(a.real) && !Double.isNaN(b.real) ,"Real At ("+i+","+j+") A = "+a.real+" B = "+b.real);
assertTrue(!Double.isInfinite(a.real) && !Double.isInfinite(b.real) ,"Real At ("+i+","+j+") A = "+a.real+" B = "+b.real);
assertTrue(Math.abs( a.real-b.real) <= tol,"Real At ("+i+","+j+") A = "+a.real+" B = "+b.real);
assertTrue(!Double.isNaN(a.imaginary) && !Double.isNaN(b.imaginary) ,"Img At ("+i+","+j+") A = "+a.imaginary+" B = "+b.imaginary);
assertTrue(!Double.isInfinite(a.imaginary) && !Double.isInfinite(b.imaginary) ,"Img At ("+i+","+j+") A = "+a.imaginary+" B = "+b.imaginary);
assertTrue(Math.abs( a.imaginary-b.imaginary) <= tol,"Img At ("+i+","+j+") A = "+a.imaginary+" B = "+b.imaginary);
}
}
}
/**
*
* Checks to see if the transpose of B is equal to A and countable:
*
*
*
* |aij - bji| ≤ tol
*
*
*
* The test will fail if any element in either matrix is NaN or infinite.
*
*
* @param A Matrix A
* @param B Matrix B
* @param tol Tolerance
*/
public static void assertEqualsTrans( RealMatrix64F A , RealMatrix64F B , double tol ) {
assertShape(A,B.getNumCols(),B.getNumRows());
for( int i = 0; i < A.getNumRows(); i++ ){
for( int j = 0; j < A.getNumCols(); j++ ) {
double valA = A.get(i,j);
double valB = B.get(j,i);
assertTrue(!Double.isNaN(valA) && !Double.isNaN(valB) ,"A("+i+","+j+") = "+valA+") B("+j+","+i+") = "+valB);
assertTrue(!Double.isInfinite(valA) && !Double.isInfinite(valB) ,"A("+i+","+j+") = "+valA+") B("+j+","+i+") = "+valB);
assertTrue(Math.abs( valA-valB) <= tol,"A("+i+","+j+") = "+valA+") B("+j+","+i+") = "+valB);
}
}
}
@SuppressWarnings({"ConstantConditions"})
private static void assertTrue( boolean result , String message ) {
// if turned on use asserts
assert result : message;
// otherwise throw an exception
if( !result ) throw new TestException(message);
}
public static class TestException extends RuntimeException {
public TestException(String message) {
super(message);
}
}
}
Ads help maintain this website.