Reformat to use with pytest

Author: chahak13

benchmarks/2d/__init__.py

@@ -3,28 +3,30 @@
 import jax.numpy as jnp
 from diffmpm import MPM
 
-curr_filepath = Path(__file__).absolute()
-curr_dir = curr_filepath.parent
-os.chdir(curr_dir)
-mpm = MPM("mpm-nodal-forces.toml")
-mpm.solve()
 
-result = jnp.load("results/uniaxial-nodal-forces/particles_0300.npz")
-## Step 300
-assert jnp.round(result["stress"][0, :, 0].min() - 0.5925210678182377, 6) == 0.0
-assert jnp.round(result["stress"][0, :, 0].max() - 0.5974539476363379, 6) == 0.0
+def test_benchmarks():
+    curr_filepath = Path(__file__).absolute()
+    curr_dir = curr_filepath.parent
+    os.chdir(curr_dir)
+    mpm = MPM("mpm-nodal-forces.toml")
+    mpm.solve()
 
-## Step 510
-result = jnp.load("results/uniaxial-nodal-forces/particles_0510.npz")
-assert jnp.round(result["stress"][0, :, 0].min() - 1.0026665338366039, 5) == 0.0
-assert jnp.round(result["stress"][0, :, 0].max() - 1.0112231542459431, 5) == 0.0
+    result = jnp.load("results/uniaxial-nodal-forces/particles_0300.npz")
+    ## Step 300
+    assert jnp.round(result["stress"][0, :, 0].min() - 0.5925210678182377, 6) == 0.0
+    assert jnp.round(result["stress"][0, :, 0].max() - 0.5974539476363379, 6) == 0.0
 
-## Step 750
-result = jnp.load("results/uniaxial-nodal-forces/particles_0750.npz")
-assert jnp.round(result["stress"][0, :, 0].min() - 1.0000053000532143, 5) == 0.0
-assert jnp.round(result["stress"][0, :, 0].max() - 1.0000225119807862, 5) == 0.0
+    ## Step 510
+    result = jnp.load("results/uniaxial-nodal-forces/particles_0510.npz")
+    assert jnp.round(result["stress"][0, :, 0].min() - 1.0026665338366039, 5) == 0.0
+    assert jnp.round(result["stress"][0, :, 0].max() - 1.0112231542459431, 5) == 0.0
 
-## Step 990
-result = jnp.load("results/uniaxial-nodal-forces/particles_0990.npz")
-assert jnp.round(result["stress"][0, :, 0].min() - 0.9999990078443788, 5) == 0.0
-assert jnp.round(result["stress"][0, :, 0].max() - 0.9999990292713694, 5) == 0.0
+    ## Step 750
+    result = jnp.load("results/uniaxial-nodal-forces/particles_0750.npz")
+    assert jnp.round(result["stress"][0, :, 0].min() - 1.0000053000532143, 5) == 0.0
+    assert jnp.round(result["stress"][0, :, 0].max() - 1.0000225119807862, 5) == 0.0
+
+    ## Step 990
+    result = jnp.load("results/uniaxial-nodal-forces/particles_0990.npz")
+    assert jnp.round(result["stress"][0, :, 0].min() - 0.9999990078443788, 5) == 0.0
+    assert jnp.round(result["stress"][0, :, 0].max() - 0.9999990292713694, 5) == 0.0

benchmarks/2d/uniaxial_nodal_forces/__init__.py

@@ -3,27 +3,29 @@
 import jax.numpy as jnp
 from diffmpm import MPM
 
-curr_filepath = Path(__file__).absolute()
-curr_dir = curr_filepath.parent
-os.chdir(curr_dir)
-mpm = MPM("mpm-particle-traction.toml")
-mpm.solve()
-result = jnp.load("results/uniaxial-particle-traction/particles_0300.npz")
-## Step 300
-assert jnp.round(result["stress"][0, :, 0].min() - 0.4450086768966724, 5) == 0.0
-assert jnp.round(result["stress"][0, :, 0].max() - 0.5966527842046769, 5) == 0.0
 
-## Step 510
-result = jnp.load("results/uniaxial-particle-traction/particles_0510.npz")
-assert jnp.round(result["stress"][0, :, 0].min() - 0.7528092313640623, 5) == 0.0
-assert jnp.round(result["stress"][0, :, 0].max() - 1.0109599915279937, 5) == 0.0
+def test_benchmarks():
+    curr_filepath = Path(__file__).absolute()
+    curr_dir = curr_filepath.parent
+    os.chdir(curr_dir)
+    mpm = MPM("mpm-particle-traction.toml")
+    mpm.solve()
+    result = jnp.load("results/uniaxial-particle-traction/particles_0300.npz")
+    ## Step 300
+    assert jnp.round(result["stress"][0, :, 0].min() - 0.4450086768966724, 5) == 0.0
+    assert jnp.round(result["stress"][0, :, 0].max() - 0.5966527842046769, 5) == 0.0
 
-## Step 750
-result = jnp.load("results/uniaxial-particle-traction/particles_0750.npz")
-assert jnp.round(result["stress"][0, :, 0].min() - 0.7500090055681591, 5) == 0.0
-assert jnp.round(result["stress"][0, :, 0].max() - 1.0000224746314728, 5) == 0.0
+    ## Step 510
+    result = jnp.load("results/uniaxial-particle-traction/particles_0510.npz")
+    assert jnp.round(result["stress"][0, :, 0].min() - 0.7528092313640623, 5) == 0.0
+    assert jnp.round(result["stress"][0, :, 0].max() - 1.0109599915279937, 5) == 0.0
 
-## Step 990
-result = jnp.load("results/uniaxial-particle-traction/particles_0990.npz")
-assert jnp.round(result["stress"][0, :, 0].min() - 0.750002924022295, 5) == 0.0
-assert jnp.round(result["stress"][0, :, 0].max() - 0.9999997782938734, 5) == 0.0
+    ## Step 750
+    result = jnp.load("results/uniaxial-particle-traction/particles_0750.npz")
+    assert jnp.round(result["stress"][0, :, 0].min() - 0.7500090055681591, 5) == 0.0
+    assert jnp.round(result["stress"][0, :, 0].max() - 1.0000224746314728, 5) == 0.0
+
+    ## Step 990
+    result = jnp.load("results/uniaxial-particle-traction/particles_0990.npz")
+    assert jnp.round(result["stress"][0, :, 0].min() - 0.750002924022295, 5) == 0.0
+    assert jnp.round(result["stress"][0, :, 0].max() - 0.9999997782938734, 5) == 0.0

benchmarks/2d/uniaxial_nodal_forces/test_benchmark.py

@@ -3,15 +3,17 @@
 import jax.numpy as jnp
 from diffmpm import MPM
 
-curr_filepath = Path(__file__).absolute()
-curr_dir = curr_filepath.parent
-os.chdir(curr_dir)
-mpm = MPM("mpm-uniaxial-stress.toml")
-mpm.solve()
 
-result = jnp.load("results/uniaxial-stress-usf/particles_09.npz")
-true_stress_yy = -1.0
-true_stress_xx = 0.0
+def test_benchmarks():
+    curr_filepath = Path(__file__).absolute()
+    curr_dir = curr_filepath.parent
+    os.chdir(curr_dir)
+    mpm = MPM("mpm-uniaxial-stress.toml")
+    mpm.solve()
 
-assert jnp.round(result["stress"][0, :, 1].max() - true_stress_yy, 8) == 0.0
-assert jnp.round(result["stress"][0, :, 0].max() - true_stress_xx, 8) == 0.0
+    result = jnp.load("results/uniaxial-stress-usf/particles_09.npz")
+    true_stress_yy = -1.0
+    true_stress_xx = 0.0
+
+    assert jnp.round(result["stress"][0, :, 1].max() - true_stress_yy, 8) == 0.0
+    assert jnp.round(result["stress"][0, :, 0].max() - true_stress_xx, 8) == 0.0

benchmarks/2d/uniaxial_particle_traction/__init__.py

@@ -0,0 +1 @@
+

benchmarks/2d/uniaxial_particle_traction/test_benchmark.py

@@ -0,0 +1 @@
+