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

condensed add_table test

This commit is contained in:
Philip Eisenlohr 2022-02-15 16:42:17 -05:00
parent 6916160ed1
commit e2d1ffe7b8
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
python/tests/test_VTK.py
View File

@ -7,7 +7,6 @@ import numpy as np
import numpy.ma as ma import numpy.ma as ma
import vtk import vtk
from collections import defaultdict
from damask import VTK from damask import VTK
from damask import Table from damask import Table
from damask import grid_filters from damask import grid_filters
@ -159,12 +158,11 @@ class TestVTK:
{'tensor':(3,3),'scalar':(1,)}]) {'tensor':(3,3),'scalar':(1,)}])
def test_add_table(self,default,shapes): def test_add_table(self,default,shapes):
N = np.random.choice([default.N_points,default.N_cells]) N = np.random.choice([default.N_points,default.N_cells])
d = defaultdict(dict) d = dict()
for k,s in shapes.items(): for k,s in shapes.items():
d[k]['shape'] = s d[k] = dict(shape = s,
d[k]['data'] = np.random.random(N*np.prod(s)).reshape((N,-1)) data = np.random.random(N*np.prod(s)).reshape((N,-1)))
t = Table(np.column_stack([d[k]['data'] for k in shapes.keys()]),shapes) default.add(Table(np.column_stack([d[k]['data'] for k in shapes.keys()]),shapes))
default.add(t)
for k,s in shapes.items(): for k,s in shapes.items():
assert np.allclose(default.get(k).reshape((N,-1)),d[k]['data'], assert np.allclose(default.get(k).reshape((N,-1)),d[k]['data'],
rtol=1e-7) rtol=1e-7)