Add test for particle volume computation

chahak13 · chahak13 · commit 7a28a38c1915 · 2023-07-01T17:14:50.000-07:00
diff --git a/tests/test_particle.py b/tests/test_particle.py
@@ -43,3 +43,7 @@ def test_compute_strain(self, elements, particles):
             ),
         )
         assert jnp.allclose(particles.volumetric_strain_centroid, jnp.array([0.2]))
+
+    def test_compute_volume(self, elements, particles):
+        particles.compute_volume(elements, elements.total_elements)
+        assert jnp.allclose(particles.volume, jnp.array([0.5, 0.5]).reshape(2, 1, 1))

Original file line number	Diff line number	Diff line change
`@@ -43,3 +43,7 @@ def test_compute_strain(self, elements, particles):`
`43`	`43`	`),`
`44`	`44`	`)`
`45`	`45`	`assert jnp.allclose(particles.volumetric_strain_centroid, jnp.array([0.2]))`
	`46`	`+`
	`47`	`+ def test_compute_volume(self, elements, particles):`
	`48`	`+ particles.compute_volume(elements, elements.total_elements)`
	`49`	`+ assert jnp.allclose(particles.volume, jnp.array([0.5, 0.5]).reshape(2, 1, 1))`