trikitrok avatar trikitrok committed 1cad6ca

refactored regression test for a short complete simulation

Comments (0)

Files changed (1)

tests/test_edmd.py

     box_size = 10
     sim = edmd.create_simulation(num_particles, seed, radius, box_size, eps, minimium_velocity, maximum_velocity)
     sim.run(10)
-    assert(np.all(np.loadtxt("tests/data/coords_regression.txt") == [particle.x for particle in sim.particles]))
-    assert(np.all(np.loadtxt("tests/data/vels_regression.txt") == [particle.v for particle in sim.particles]))
-    assert(np.all(np.loadtxt("tests/data/t_events_regression.txt") == sim.t_events))
+    tol = 1.0e-8
+    expected_coords = np.loadtxt("tests/data/coords_regression.txt")
+    expected_vels = np.loadtxt("tests/data/vels_regression.txt")
+    expected_t_events = np.loadtxt("tests/data/t_events_regression.txt")
+    assert(np.all( np.abs(expected_coords - [particle.x for particle in sim.particles] ) < tol))
+    assert(np.all( np.abs(expected_vels - [particle.v for particle in sim.particles] ) < tol))
+    assert(np.all( np.abs(expected_t_events - sim.t_events) < tol ) )
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.