issue in test_mechanics
This commit is contained in:
parent
694b7ec3c5
commit
36f04309b4
|
@ -150,8 +150,8 @@ class TestMechanics:
|
||||||
def test_vectorize_2_arg(self,vectorized,single):
|
def test_vectorize_2_arg(self,vectorized,single):
|
||||||
P = np.random.rand(self.n,3,3)
|
P = np.random.rand(self.n,3,3)
|
||||||
F = np.random.rand(self.n,3,3)
|
F = np.random.rand(self.n,3,3)
|
||||||
P_vec = np.random.rand(self.n//10,10,3,3)
|
P_vec = np.reshape(P,(self.n//10,10,3,3))
|
||||||
F_vec = np.random.rand(self.n//10,10,3,3)
|
F_vec = np.reshape(F,(self.n//10,10,3,3))
|
||||||
for i,v in enumerate(np.reshape(vectorized(P_vec,F_vec),vectorized(P,F).shape)):
|
for i,v in enumerate(np.reshape(vectorized(P_vec,F_vec),vectorized(P,F).shape)):
|
||||||
assert np.allclose(single(P[i],F[i]),v)
|
assert np.allclose(single(P[i],F[i]),v)
|
||||||
|
|
||||||
|
@ -159,7 +159,7 @@ class TestMechanics:
|
||||||
@pytest.mark.parametrize('vectorized,single',[(mechanics.strain_tensor,strain_tensor)])
|
@pytest.mark.parametrize('vectorized,single',[(mechanics.strain_tensor,strain_tensor)])
|
||||||
def test_vectorize_strain_tensor(self,vectorized,single):
|
def test_vectorize_strain_tensor(self,vectorized,single):
|
||||||
F = np.random.rand(self.n,3,3)
|
F = np.random.rand(self.n,3,3)
|
||||||
F_vec = np.random.rand(self.n//10,10,3,3)
|
F_vec = np.reshape(F,(self.n//10,10,3,3))
|
||||||
t = ['V','U'][np.random.randint(0,2)]
|
t = ['V','U'][np.random.randint(0,2)]
|
||||||
m = np.random.random()*10.0 -5.0
|
m = np.random.random()*10.0 -5.0
|
||||||
for i,v in enumerate(np.reshape(vectorized(F_vec,t,m),vectorized(F,t,m).shape)):
|
for i,v in enumerate(np.reshape(vectorized(F_vec,t,m),vectorized(F,t,m).shape)):
|
||||||
|
|
Loading…
Reference in New Issue