223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

testing 2-dim array of tensors

This commit is contained in:
Martin Diehl 2020-05-27 21:07:48 +02:00
parent 56afc03f3c
commit 00168a5939
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
python/tests/test_mechanics.py
View File

@ -20,9 +20,10 @@ class TestMechanics:
(mechanics.spherical_part, spherical_part)
])
def test_vectorize_1_arg_(self,vectorized,single):
test_data = np.random.rand(self.n,3,3)
for i,v in enumerate(vectorized(test_data)):
assert np.allclose(single(test_data[i]),v)
test_data_flat = np.random.rand(self.n,3,3)
test_data = np.reshape(test_data_flat,(self.n//10,10,3,3))
for i,v in enumerate(np.reshape(vectorized(test_data),vectorized(test_data_flat).shape)):
assert np.allclose(single(test_data_flat[i]),v)
@pytest.mark.parametrize('function',[mechanics.deviatoric_part,
mechanics.eigenvalues,