Issue #163: Linting and docs

SciKit-Surgery · Jan 14, 2021 · d07acc2 · d07acc2
1 parent 47353a6
commit d07acc2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/docs/module_ref.rst b/docs/module_ref.rst
@@ -58,6 +58,12 @@ Surface Models
    :undoc-members:
    :show-inheritance:
 
+Unstructured Grid Model
+^^^^^^^^^^^^^^^^^^^^^^^
+.. automodule:: sksurgeryvtk.models.vtk_grid_model
+   :members:
+   :undoc-members:
+   :show-inheritance:
 Image Model
 ^^^^^^^^^^^
 .. automodule:: sksurgeryvtk.models.vtk_image_model

diff --git a/tests/models/test_vtk_unstructured_grid.py b/tests/models/test_vtk_unstructured_grid.py
@@ -5,15 +5,15 @@
 def test_load_from_vtu():
 
     model = "tests/data/models/unstructured_grid.vtu"
-    neonate_model = GM.vtkUnstructuredGridModel(model)
+    neonate_model = GM.VTKUnstructuredGridModel(model)
 
     array_min, array_max = neonate_model.get_cell_array_bounds()
     neonate_model.threshold_between(array_min, array_max)
 
 def test_load_from_vtk():
 
     model = "tests/data/models/unstructured_grid.vtk"
-    neonate_model = GM.vtkUnstructuredGridModel(model)
+    neonate_model = GM.VTKUnstructuredGridModel(model)
 
     array_min, array_max = neonate_model.get_cell_array_bounds()
     neonate_model.threshold_between(array_min, array_max)