@@ -10,11 +10,17 @@ import org.jetbrains.kotlinx.multik.ndarray.operations.*
1010import java.io.File
1111import kotlin.math.PI
1212import kotlin.math.roundToInt
13+ import kotlin.test.BeforeTest
1314import kotlin.test.Test
1415import kotlin.test.assertEquals
1516
1617class ModernRoboticsTestJvm {
1718
19+ @BeforeTest
20+ fun setup () {
21+ mk.setEngine(KEEngineType )
22+ }
23+
1824 @Test
1925 fun `test InverseDynamicsTrajectory` () {
2026
@@ -77,9 +83,8 @@ class ModernRoboticsTestJvm {
7783 ]).transpose(),
7884 )
7985 assertEquals(
80- mk.d2arrayFromFile(" InverseDynamicsTrajectory.csv" ),
81- taumat,
82- " $taumat not equal ${mk.d2arrayFromFile(" InverseDynamicsTrajectory.csv" )} "
86+ mk.d2arrayFromFile(" InverseDynamicsTrajectory.csv" ).round(10 ),
87+ taumat.round(10 )
8388 )
8489
8590 // Output using kandy to plot the joint forces/torques
@@ -139,13 +144,12 @@ class ModernRoboticsTestJvm {
139144 intRes = 8
140145 )
141146 assertEquals(
142- mk.d2arrayFromFile(" ForwardDynamicsTrajectory-Theta.csv" ),
143- thetamat,
144- " $thetamat not equal ${mk.d2arrayFromFile(" ForwardDynamicsTrajectory-Theta.csv" )} "
147+ mk.d2arrayFromFile(" ForwardDynamicsTrajectory-Theta.csv" ).round(10 ),
148+ thetamat.round(10 )
145149 )
146150 assertEquals(
147- mk.d2arrayFromFile(" ForwardDynamicsTrajectory-DTheta.csv" ),
148- dthetamat
151+ mk.d2arrayFromFile(" ForwardDynamicsTrajectory-DTheta.csv" ).round( 10 ) ,
152+ dthetamat.round( 10 )
149153 )
150154
151155 // Output using kandy to plot the joint angle/velocities
@@ -263,12 +267,12 @@ class ModernRoboticsTestJvm {
263267 intRes
264268 )
265269 assertEquals(
266- mk.d2arrayFromFile(" SimulateControl-ThetaActual.csv" ),
267- thetamat
270+ mk.d2arrayFromFile(" SimulateControl-ThetaActual.csv" ).round( 10 ) ,
271+ thetamat.round( 10 )
268272 )
269273 assertEquals(
270- mk.d2arrayFromFile(" SimulateControl-ThetaDesired.csv" ),
271- thetamatd
274+ mk.d2arrayFromFile(" SimulateControl-ThetaDesired.csv" ).round( 10 ) ,
275+ thetamatd.round( 10 )
272276 )
273277
274278 // Output using kandy to plot
0 commit comments