File: TestAMRSliceFilter.py

package info (click to toggle)
vtk7 7.1.1%2Bdfsg1-12
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 125,776 kB
  • sloc: cpp: 1,539,582; ansic: 106,521; python: 78,038; tcl: 47,013; xml: 8,142; yacc: 5,040; java: 4,439; perl: 3,132; lex: 1,926; sh: 1,500; makefile: 122; objc: 83
file content (41 lines) | stat: -rwxr-xr-x 1,088 bytes parent folder | download | duplicates (11)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
#!/usr/bin/env python

# This tests vtkAMRResampleFilter

import vtk
from vtk.test import Testing
from vtk.util.misc import vtkGetDataRoot

VTK_DATA_ROOT = vtkGetDataRoot()

def NumCells(out):
  n =0;
  for i in range(out.GetNumberOfLevels()):
    for j in range(out.GetNumberOfDataSets(i)):
      m = out.GetDataSet(i,j).GetNumberOfCells()
      #print (i,j,m)
      n = n+ m
  return n

class TestAMRSliceFilter(Testing.vtkTest):
  def testMe(self):
    filename= VTK_DATA_ROOT +"/Data/AMR/Enzo/DD0010/moving7_0010.hierarchy"
    datafieldname = "TotalEnergy"

    reader = vtk.vtkAMREnzoReader()
    reader.SetFileName(filename);
    reader.SetMaxLevel(10);
    reader.SetCellArrayStatus(datafieldname,1)

    filter = vtk.vtkAMRSliceFilter()
    filter.SetInputConnection(reader.GetOutputPort())
    filter.SetNormal(1);
    filter.SetOffSetFromOrigin(0.5);
    filter.SetMaxResolution(10);
    filter.Update()
    out = filter.GetOutputDataObject(0);
    self.assertEqual(NumCells(out),456)
    out.Audit();

if __name__ == "__main__":
  Testing.main([(TestAMRSliceFilter, 'test')])