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