/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.apache.commons.math4.linear;
import org.apache.commons.math4.exception.MathIllegalArgumentException;
import org.apache.commons.math4.linear.ArrayRealVector;
import org.apache.commons.math4.linear.RealVector;
import org.junit.Assert;
import org.junit.Test;
/**
* Test cases for the {@link ArrayRealVector} class.
*
*/
public class ArrayRealVectorTest extends RealVectorAbstractTest {
@Override
public RealVector create(final double[] data) {
return new ArrayRealVector(data, true);
}
@Test
public void testConstructors() {
final double[] vec1 = {1d, 2d, 3d};
final double[] vec3 = {7d, 8d, 9d};
final double[] vec4 = {1d, 2d, 3d, 4d, 5d, 6d, 7d, 8d, 9d};
final Double[] dvec1 = {1d, 2d, 3d, 4d, 5d, 6d, 7d, 8d, 9d};
ArrayRealVector v0 = new ArrayRealVector();
Assert.assertEquals("testData len", 0, v0.getDimension());
ArrayRealVector v1 = new ArrayRealVector(7);
Assert.assertEquals("testData len", 7, v1.getDimension());
Assert.assertEquals("testData is 0.0 ", 0.0, v1.getEntry(6), 0);
ArrayRealVector v2 = new ArrayRealVector(5, 1.23);
Assert.assertEquals("testData len", 5, v2.getDimension());
Assert.assertEquals("testData is 1.23 ", 1.23, v2.getEntry(4), 0);
ArrayRealVector v3 = new ArrayRealVector(vec1);
Assert.assertEquals("testData len", 3, v3.getDimension());
Assert.assertEquals("testData is 2.0 ", 2.0, v3.getEntry(1), 0);
ArrayRealVector v3_bis = new ArrayRealVector(vec1, true);
Assert.assertEquals("testData len", 3, v3_bis.getDimension());
Assert.assertEquals("testData is 2.0 ", 2.0, v3_bis.getEntry(1), 0);
Assert.assertNotSame(v3_bis.getDataRef(), vec1);
Assert.assertNotSame(v3_bis.toArray(), vec1);
ArrayRealVector v3_ter = new ArrayRealVector(vec1, false);
Assert.assertEquals("testData len", 3, v3_ter.getDimension());
Assert.assertEquals("testData is 2.0 ", 2.0, v3_ter.getEntry(1), 0);
Assert.assertSame(v3_ter.getDataRef(), vec1);
Assert.assertNotSame(v3_ter.toArray(), vec1);
ArrayRealVector v4 = new ArrayRealVector(vec4, 3, 2);
Assert.assertEquals("testData len", 2, v4.getDimension());
Assert.assertEquals("testData is 4.0 ", 4.0, v4.getEntry(0), 0);
try {
new ArrayRealVector(vec4, 8, 3);
Assert.fail("MathIllegalArgumentException expected");
} catch (MathIllegalArgumentException ex) {
// expected behavior
}
RealVector v5_i = new ArrayRealVector(dvec1);
Assert.assertEquals("testData len", 9, v5_i.getDimension());
Assert.assertEquals("testData is 9.0 ", 9.0, v5_i.getEntry(8), 0);
ArrayRealVector v5 = new ArrayRealVector(dvec1);
Assert.assertEquals("testData len", 9, v5.getDimension());
Assert.assertEquals("testData is 9.0 ", 9.0, v5.getEntry(8), 0);
ArrayRealVector v6 = new ArrayRealVector(dvec1, 3, 2);
Assert.assertEquals("testData len", 2, v6.getDimension());
Assert.assertEquals("testData is 4.0 ", 4.0, v6.getEntry(0), 0);
try {
new ArrayRealVector(dvec1, 8, 3);
Assert.fail("MathIllegalArgumentException expected");
} catch (MathIllegalArgumentException ex) {
// expected behavior
}
ArrayRealVector v7 = new ArrayRealVector(v1);
Assert.assertEquals("testData len", 7, v7.getDimension());
Assert.assertEquals("testData is 0.0 ", 0.0, v7.getEntry(6), 0);
RealVectorTestImpl v7_i = new RealVectorTestImpl(vec1);
ArrayRealVector v7_2 = new ArrayRealVector(v7_i);
Assert.assertEquals("testData len", 3, v7_2.getDimension());
Assert.assertEquals("testData is 0.0 ", 2.0d, v7_2.getEntry(1), 0);
ArrayRealVector v8 = new ArrayRealVector(v1, true);
Assert.assertEquals("testData len", 7, v8.getDimension());
Assert.assertEquals("testData is 0.0 ", 0.0, v8.getEntry(6), 0);
Assert.assertNotSame("testData not same object ", v1.getDataRef(), v8.getDataRef());
ArrayRealVector v8_2 = new ArrayRealVector(v1, false);
Assert.assertEquals("testData len", 7, v8_2.getDimension());
Assert.assertEquals("testData is 0.0 ", 0.0, v8_2.getEntry(6), 0);
Assert.assertEquals("testData same object ", v1.getDataRef(), v8_2.getDataRef());
ArrayRealVector v9 = new ArrayRealVector(v1, v3);
Assert.assertEquals("testData len", 10, v9.getDimension());
Assert.assertEquals("testData is 1.0 ", 1.0, v9.getEntry(7), 0);
ArrayRealVector v10 = new ArrayRealVector(v2, new RealVectorTestImpl(vec3));
Assert.assertEquals("testData len", 8, v10.getDimension());
Assert.assertEquals("testData is 1.23 ", 1.23, v10.getEntry(4), 0);
Assert.assertEquals("testData is 7.0 ", 7.0, v10.getEntry(5), 0);
ArrayRealVector v11 = new ArrayRealVector(new RealVectorTestImpl(vec3), v2);
Assert.assertEquals("testData len", 8, v11.getDimension());
Assert.assertEquals("testData is 9.0 ", 9.0, v11.getEntry(2), 0);
Assert.assertEquals("testData is 1.23 ", 1.23, v11.getEntry(3), 0);
ArrayRealVector v12 = new ArrayRealVector(v2, vec3);
Assert.assertEquals("testData len", 8, v12.getDimension());
Assert.assertEquals("testData is 1.23 ", 1.23, v12.getEntry(4), 0);
Assert.assertEquals("testData is 7.0 ", 7.0, v12.getEntry(5), 0);
ArrayRealVector v13 = new ArrayRealVector(vec3, v2);
Assert.assertEquals("testData len", 8, v13.getDimension());
Assert.assertEquals("testData is 9.0 ", 9.0, v13.getEntry(2), 0);
Assert.assertEquals("testData is 1.23 ", 1.23, v13.getEntry(3), 0);
ArrayRealVector v14 = new ArrayRealVector(vec3, vec4);
Assert.assertEquals("testData len", 12, v14.getDimension());
Assert.assertEquals("testData is 9.0 ", 9.0, v14.getEntry(2), 0);
Assert.assertEquals("testData is 1.0 ", 1.0, v14.getEntry(3), 0);
}
@Test
public void testGetDataRef() {
final double[] data = {1d, 2d, 3d, 4d};
final ArrayRealVector v = new ArrayRealVector(data);
v.getDataRef()[0] = 0d;
Assert.assertEquals("", 0d, v.getEntry(0), 0);
}
@Test
public void testPredicates() {
Assert.assertEquals(create(new double[] { Double.NaN, 1, 2 }).hashCode(),
create(new double[] { 0, Double.NaN, 2 }).hashCode());
Assert.assertTrue(create(new double[] { Double.NaN, 1, 2 }).hashCode() !=
create(new double[] { 0, 1, 2 }).hashCode());
}
@Test
public void testZeroVectors() {
Assert.assertEquals(0, new ArrayRealVector(new double[0]).getDimension());
Assert.assertEquals(0, new ArrayRealVector(new double[0], true).getDimension());
Assert.assertEquals(0, new ArrayRealVector(new double[0], false).getDimension());
}
}