complaints from prospector (PEP8)
This commit is contained in:
parent
63465f4d27
commit
6dcf6b972c
|
@ -287,7 +287,7 @@ class TestOrientation:
|
||||||
@pytest.mark.parametrize('family',crystal_families)
|
@pytest.mark.parametrize('family',crystal_families)
|
||||||
@pytest.mark.parametrize('proper',[True,False])
|
@pytest.mark.parametrize('proper',[True,False])
|
||||||
def test_in_SST(self,family,proper):
|
def test_in_SST(self,family,proper):
|
||||||
assert Orientation(family=family).in_SST(np.zeros(3),proper)
|
assert Orientation(family=family).in_SST(np.zeros(3),proper)
|
||||||
|
|
||||||
@pytest.mark.parametrize('function',['in_SST','IPF_color'])
|
@pytest.mark.parametrize('function',['in_SST','IPF_color'])
|
||||||
def test_invalid_argument(self,function):
|
def test_invalid_argument(self,function):
|
||||||
|
|
|
@ -367,13 +367,13 @@ class TestResult:
|
||||||
|
|
||||||
@pytest.mark.parametrize('mode',['cell','node'])
|
@pytest.mark.parametrize('mode',['cell','node'])
|
||||||
def test_coordinates(self,default,mode):
|
def test_coordinates(self,default,mode):
|
||||||
if mode == 'cell':
|
if mode == 'cell':
|
||||||
a = grid_filters.coordinates0_point(default.cells,default.size,default.origin)
|
a = grid_filters.coordinates0_point(default.cells,default.size,default.origin)
|
||||||
b = default.coordinates0_point.reshape(tuple(default.cells)+(3,),order='F')
|
b = default.coordinates0_point.reshape(tuple(default.cells)+(3,),order='F')
|
||||||
elif mode == 'node':
|
elif mode == 'node':
|
||||||
a = grid_filters.coordinates0_node(default.cells,default.size,default.origin)
|
a = grid_filters.coordinates0_node(default.cells,default.size,default.origin)
|
||||||
b = default.coordinates0_node.reshape(tuple(default.cells+1)+(3,),order='F')
|
b = default.coordinates0_node.reshape(tuple(default.cells+1)+(3,),order='F')
|
||||||
assert np.allclose(a,b)
|
assert np.allclose(a,b)
|
||||||
|
|
||||||
@pytest.mark.parametrize('output',['F','*',['P'],['P','F']],ids=range(4))
|
@pytest.mark.parametrize('output',['F','*',['P'],['P','F']],ids=range(4))
|
||||||
@pytest.mark.parametrize('fname',['12grains6x7x8_tensionY.hdf5'],ids=range(1))
|
@pytest.mark.parametrize('fname',['12grains6x7x8_tensionY.hdf5'],ids=range(1))
|
||||||
|
@ -421,7 +421,7 @@ class TestResult:
|
||||||
def test_XDMF_datatypes(self,tmp_path,single_phase,update,ref_path):
|
def test_XDMF_datatypes(self,tmp_path,single_phase,update,ref_path):
|
||||||
for shape in [('scalar',()),('vector',(3,)),('tensor',(3,3)),('matrix',(12,))]:
|
for shape in [('scalar',()),('vector',(3,)),('tensor',(3,3)),('matrix',(12,))]:
|
||||||
for dtype in ['f4','f8','i1','i2','i4','i8','u1','u2','u4','u8']:
|
for dtype in ['f4','f8','i1','i2','i4','i8','u1','u2','u4','u8']:
|
||||||
single_phase.add_calculation(f"np.ones(np.shape(#F#)[0:1]+{shape[1]},'{dtype}')",f'{shape[0]}_{dtype}')
|
single_phase.add_calculation(f"np.ones(np.shape(#F#)[0:1]+{shape[1]},'{dtype}')",f'{shape[0]}_{dtype}')
|
||||||
fname = os.path.splitext(os.path.basename(single_phase.fname))[0]+'.xdmf'
|
fname = os.path.splitext(os.path.basename(single_phase.fname))[0]+'.xdmf'
|
||||||
os.chdir(tmp_path)
|
os.chdir(tmp_path)
|
||||||
single_phase.export_XDMF()
|
single_phase.export_XDMF()
|
||||||
|
|
|
@ -1076,19 +1076,19 @@ class TestRotation:
|
||||||
def test_from_fiber_component(self,N,sigma):
|
def test_from_fiber_component(self,N,sigma):
|
||||||
p = []
|
p = []
|
||||||
for run in range(5):
|
for run in range(5):
|
||||||
alpha = np.random.random()*2*np.pi,np.arccos(np.random.random())
|
alpha = np.random.random()*2*np.pi,np.arccos(np.random.random())
|
||||||
beta = np.random.random()*2*np.pi,np.arccos(np.random.random())
|
beta = np.random.random()*2*np.pi,np.arccos(np.random.random())
|
||||||
|
|
||||||
f_in_C = np.array([np.sin(alpha[0])*np.cos(alpha[1]), np.sin(alpha[0])*np.sin(alpha[1]), np.cos(alpha[0])])
|
f_in_C = np.array([np.sin(alpha[0])*np.cos(alpha[1]), np.sin(alpha[0])*np.sin(alpha[1]), np.cos(alpha[0])])
|
||||||
f_in_S = np.array([np.sin(beta[0] )*np.cos(beta[1] ), np.sin(beta[0] )*np.sin(beta[1] ), np.cos(beta[0] )])
|
f_in_S = np.array([np.sin(beta[0] )*np.cos(beta[1] ), np.sin(beta[0] )*np.sin(beta[1] ), np.cos(beta[0] )])
|
||||||
ax = np.append(np.cross(f_in_C,f_in_S), - np.arccos(np.dot(f_in_C,f_in_S)))
|
ax = np.append(np.cross(f_in_C,f_in_S), - np.arccos(np.dot(f_in_C,f_in_S)))
|
||||||
n = Rotation.from_axis_angle(ax if ax[3] > 0.0 else ax*-1.0 ,normalize=True) # rotation to align fiber axis in crystal and sample system
|
n = Rotation.from_axis_angle(ax if ax[3] > 0.0 else ax*-1.0 ,normalize=True) # rotation to align fiber axis in crystal and sample system
|
||||||
|
|
||||||
o = Rotation.from_fiber_component(alpha,beta,np.radians(sigma),N,False)
|
o = Rotation.from_fiber_component(alpha,beta,np.radians(sigma),N,False)
|
||||||
angles = np.arccos(np.clip(np.dot(o@np.broadcast_to(f_in_S,(N,3)),n@f_in_S),-1,1))
|
angles = np.arccos(np.clip(np.dot(o@np.broadcast_to(f_in_S,(N,3)),n@f_in_S),-1,1))
|
||||||
dist = np.array(angles) * (np.random.randint(0,2,N)*2-1)
|
dist = np.array(angles) * (np.random.randint(0,2,N)*2-1)
|
||||||
|
|
||||||
p.append(stats.normaltest(dist)[1])
|
p.append(stats.normaltest(dist)[1])
|
||||||
|
|
||||||
sigma_out = np.degrees(np.std(dist))
|
sigma_out = np.degrees(np.std(dist))
|
||||||
p = np.average(p)
|
p = np.average(p)
|
||||||
|
|
|
@ -173,11 +173,11 @@ class TestVTK:
|
||||||
polyData = VTK.from_poly_data(points)
|
polyData = VTK.from_poly_data(points)
|
||||||
polyData.add(points,'coordinates')
|
polyData.add(points,'coordinates')
|
||||||
if update:
|
if update:
|
||||||
polyData.save(ref_path/'polyData')
|
polyData.save(ref_path/'polyData')
|
||||||
else:
|
else:
|
||||||
reference = VTK.load(ref_path/'polyData.vtp')
|
reference = VTK.load(ref_path/'polyData.vtp')
|
||||||
assert polyData.__repr__() == reference.__repr__() and \
|
assert polyData.__repr__() == reference.__repr__() and \
|
||||||
np.allclose(polyData.get('coordinates'),points)
|
np.allclose(polyData.get('coordinates'),points)
|
||||||
|
|
||||||
@pytest.mark.xfail(int(vtk.vtkVersion.GetVTKVersion().split('.')[0])<8, reason='missing METADATA')
|
@pytest.mark.xfail(int(vtk.vtkVersion.GetVTKVersion().split('.')[0])<8, reason='missing METADATA')
|
||||||
def test_compare_reference_rectilinearGrid(self,update,ref_path,tmp_path):
|
def test_compare_reference_rectilinearGrid(self,update,ref_path,tmp_path):
|
||||||
|
@ -189,8 +189,8 @@ class TestVTK:
|
||||||
rectilinearGrid.add(np.ascontiguousarray(c),'cell')
|
rectilinearGrid.add(np.ascontiguousarray(c),'cell')
|
||||||
rectilinearGrid.add(np.ascontiguousarray(n),'node')
|
rectilinearGrid.add(np.ascontiguousarray(n),'node')
|
||||||
if update:
|
if update:
|
||||||
rectilinearGrid.save(ref_path/'rectilinearGrid')
|
rectilinearGrid.save(ref_path/'rectilinearGrid')
|
||||||
else:
|
else:
|
||||||
reference = VTK.load(ref_path/'rectilinearGrid.vtr')
|
reference = VTK.load(ref_path/'rectilinearGrid.vtr')
|
||||||
assert rectilinearGrid.__repr__() == reference.__repr__() and \
|
assert rectilinearGrid.__repr__() == reference.__repr__() and \
|
||||||
np.allclose(rectilinearGrid.get('cell'),c)
|
np.allclose(rectilinearGrid.get('cell'),c)
|
||||||
|
|
Loading…
Reference in New Issue