• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• discretize-master
  • Makefile
  • .coveragerc
  • examples
    • plot_cyl_mirror.py
    • plot_image.py
    • plot_cahn_hilliard.py
    • plot_streamThickness.py
    • plot_pyvista_laguna.py
    • plot_dc_resistivity.py
    • README.txt
    • plot_quadtree_divergence.py
    • plot_slicer_demo.py
    • plot_quadtree_hanging.py
  • LICENSE
  • appveyor.yml
  • CITATION.rst
  • README.rst
  • .bumpversion.cfg
  • setup.py
  • credentials.tar.gz.enc
  • tutorials
    • pde
      • README.txt
      • 1_poisson.py
      • 2_advection_diffusion.py
    • operators
      • 1_averaging.py
      • README.txt
      • 2_differential.py
    • inner_products
      • 3_calculus.py
      • 1_basic.py
      • README.txt
      • 2_physical_properties.py
      • 4_advanced.py
    • mesh_generation
      • 4_tree_mesh.py
      • 1_mesh_overview.py
      • README.txt
      • 3_cylindrical_mesh.py
      • 2_tensor_mesh.py
  • AUTHORS.rst
  • .travis.yml
  • discretize
    • utils
      • interputils.py
      • meshutils.py
      • io_utils.py
      • codeutils.py
      • coordutils.py
      • setup.py
      • matutils.py
      • __init__.py
      • interputils_cython.pyx
      • curvutils.py
    • TensorMesh.py
    • View.py
    • CylMesh.py
    • tree.cpp
    • base
      • base_tensor_mesh.py
      • base_mesh.py
      • __init__.py
    • DiffOperators.py
    • tree.h
    • setup.py
    • CurvilinearMesh.py
    • __init__.py
    • tree.pxd
    • MeshIO.py
    • tree_ext.pyx
    • InnerProducts.py
    • Tests.py
    • mixins
      • vtkModule.py
      • omfModule.py
      • __init__.py
    • TreeMesh.py
  • tests
    • base
      • test_coordutils.py
      • test_utils.py
      • test_view.py
      • test_tensor.py
      • test_tensor_omf.py
      • test_tensor_io.py
      • test_curvilinear.py
      • test_interpolation.py
      • test_tensor_innerproduct.py
      • test_basemesh.py
      • test_tensor_vtk.py
      • test_properties.py
      • test_curvilinear_vtk.py
      • test_tensor_innerproduct_derivs.py
      • test_operators.py
      • test_tensor_boundary.py
      • test_tensor_boundary_poisson.py
    • tree
      • test_tree_innerproduct_derivs.py
      • test_tree_utils.py
      • test_tree_vtk.py
      • test_tree_operators.py
      • test_tree_interpolation.py
      • test_tree_io.py
      • test_tree.py
    • cyl
      • test_cyl.py
      • test_cylOperators.py
      • test_cyl_innerproducts.py
      • test_cyl3D.py
    • docs
      • test_docs.py
  • requirements.txt
  • ci
    • deploy-pypi.sh
    • setup-miniconda.sh
  • .gitignore
  • docs
    • Makefile
    • make.bat
    • content
      • big_picture.rst
      • installing.rst
      • inner_products.rst
      • additional_resources.rst
      • finite_volume.rst
    • images
      • logo-block.ico
    • api
      • index.rst
    • release
      • 0.4.13-notes.rst
      • 0.4.12-notes.rst
      • 0.4.14-notes.rst
      • 0.4.15-notes.rst
      • index.rst
    • index.rst
    • conf.py
    • _templates
      • breadcrumbs.html
      • footer.html
      • autosummary
        • exception.rst
        • module.rst
        • class.rst
        • base.rst
        • function.rst
      • layout.html
  • MANIFEST.in
  • github_deploy_key_simpeg_discretize_docs.enc
  • requirements_dev.txt

from __future__ import print_function
import numpy as np
import unittest
import discretize

try:
    import vtk.util.numpy_support as nps
except ImportError:
    has_vtk = False
else:
    has_vtk = True


if has_vtk:
    class TestTreeMeshVTK(unittest.TestCase):

        def setUp(self):
            h = np.ones(16)
            mesh = discretize.TreeMesh([h, 2*h, 3*h])
            cell_points = np.array([[0.5, 0.5, 0.5],
                                    [0.5, 2.5, 0.5]])
            cell_levels = np.array([4, 4])
            mesh.insert_cells(cell_points, cell_levels)
            self.mesh = mesh

        def test_VTK_object_conversion(self):
            mesh = self.mesh
            vec = np.arange(mesh.nC)
            models = {'arange': vec}

            vtkObj = mesh.to_vtk(models)

            self.assertEqual(mesh.nC, vtkObj.GetNumberOfCells())
            # TODO: this is actually different?: self.assertEqual(mesh.nN, vtkObj.GetNumberOfPoints())
            # Remember that the tree vtk conversion adds two arrays
            self.assertEqual(len(models.keys())+2, vtkObj.GetCellData().GetNumberOfArrays())
            bnds = vtkObj.GetBounds()
            self.assertEqual(mesh.x0[0], bnds[0])
            self.assertEqual(mesh.x0[1], bnds[2])
            self.assertEqual(mesh.x0[2], bnds[4])

            names = list(models.keys())
            for name in names:
                arr = nps.vtk_to_numpy(vtkObj.GetCellData().GetArray(name))
                arr = arr.flatten(order='F')
                self.assertTrue(np.allclose(models[name], arr))



if __name__ == '__main__':
    unittest.main()