[wpimath] Make Vector-Vector binary operators return Vector (#5772)

Fixes #5741.
This commit is contained in:
Tyler Veness
2023-10-17 16:44:30 -07:00
committed by GitHub
parent ed93889e17
commit 02cbbc997d
2 changed files with 59 additions and 0 deletions

View File

@@ -9,6 +9,44 @@ import static org.junit.jupiter.api.Assertions.assertEquals;
import org.junit.jupiter.api.Test;
class VectorTest {
@Test
void testVectorPlus() {
var vec1 = VecBuilder.fill(1.0, 2.0, 3.0);
var vec2 = VecBuilder.fill(4.0, 5.0, 6.0);
var result1 = vec1.plus(vec2);
assertEquals(5.0, result1.get(0, 0));
assertEquals(7.0, result1.get(1, 0));
assertEquals(9.0, result1.get(2, 0));
var vec3 = VecBuilder.fill(-1.0, 2.0, -3.0);
var vec4 = VecBuilder.fill(4.0, -5.0, 6.0);
var result2 = vec3.plus(vec4);
assertEquals(3.0, result2.get(0, 0));
assertEquals(-3.0, result2.get(1, 0));
assertEquals(3.0, result2.get(2, 0));
}
@Test
void testVectorMinus() {
var vec1 = VecBuilder.fill(1.0, 2.0, 3.0);
var vec2 = VecBuilder.fill(4.0, 5.0, 6.0);
var result1 = vec1.minus(vec2);
assertEquals(-3.0, result1.get(0, 0));
assertEquals(-3.0, result1.get(1, 0));
assertEquals(-3.0, result1.get(2, 0));
var vec3 = VecBuilder.fill(-1.0, 2.0, -3.0);
var vec4 = VecBuilder.fill(4.0, -5.0, 6.0);
var result2 = vec3.minus(vec4);
assertEquals(-5.0, result2.get(0, 0));
assertEquals(7.0, result2.get(1, 0));
assertEquals(-9.0, result2.get(2, 0));
}
@Test
void testVectorDot() {
var vec1 = VecBuilder.fill(1.0, 2.0, 3.0);