From 7f07c08cdc45f0e17b6b72eae6317b10c458836a Mon Sep 17 00:00:00 2001
From: ellis <eshoag2@illinois.edu>
Date: Sun, 12 Feb 2017 23:51:34 -0600
Subject: [PATCH] spelling
---
meshmode/mesh/processing.py | 3 ++-
test/test_meshmode.py | 17 +++++++++--------
2 files changed, 11 insertions(+), 9 deletions(-)
diff --git a/meshmode/mesh/processing.py b/meshmode/mesh/processing.py
index 02a7e3bc..00b7a8bf 100644
--- a/meshmode/mesh/processing.py
+++ b/meshmode/mesh/processing.py
@@ -54,7 +54,8 @@ def partition_mesh(mesh, part_per_element, part_nr):
*part_to_global* is a :class:`numpy.ndarray` mapping element
numbers on *part_mesh* to ones in *mesh*.
"""
- assert len(part_per_element) == mesh.nelements, "part_per_element must have shape (mesh.nelements,)"
+ assert len(part_per_element) == mesh.nelements, (
+ "part_per_element must have shape (mesh.nelements,)")
# Contains the indices of the elements requested.
queried_elems = np.where(np.array(part_per_element) == part_nr)[0]
diff --git a/test/test_meshmode.py b/test/test_meshmode.py
index 740b917b..4cdffd85 100644
--- a/test/test_meshmode.py
+++ b/test/test_meshmode.py
@@ -50,7 +50,7 @@ logger = logging.getLogger(__name__)
# {{{ partition_mesh
-@pytest.mark.parameterize("mesh_type", ["torus", "box"])
+@pytest.mark.parametrize("mesh_type", ["torus", "box"])
def test_partition_mesh(mesh_type):
if mesh_type == "torus":
from meshmode.mesh.generation import generate_torus
@@ -59,14 +59,14 @@ def test_partition_mesh(mesh_type):
part_per_element = np.array([0, 1, 2, 1, 1, 2, 1, 0])
from meshmode.mesh.processing import partition_mesh
- (part_mesh, part_to_global) = partition_mesh(my_mesh, part_per_element, 0)
- assert part_mesh.nelements == 2
+ (part_mesh0, _) = partition_mesh(my_mesh, part_per_element, 0)
+ (part_mesh1, _) = partition_mesh(my_mesh, part_per_element, 1)
+ (part_mesh2, _) = partition_mesh(my_mesh, part_per_element, 2)
- (part_mesh, part_to_global) = partition_mesh(my_mesh, part_per_element, 1)
- assert part_mesh.nelements == 4
+ assert part_mesh0.nelements == 2
+ assert part_mesh1.nelements == 4
+ assert part_mesh2.nelements == 2
- (part_mesh, part_to_global) = partition_mesh(my_mesh, part_per_element, 2)
- assert part_mesh.nelements == 2
elif mesh_type == "box":
from meshmode.mesh.generation import generate_box_mesh
seg = np.linspace(0, 1, 10)
@@ -76,7 +76,8 @@ def test_partition_mesh(mesh_type):
adjacency_list = np.zeros((mesh.nelements,), dtype=set)
for elem in range(mesh.nelements):
adjacency_list[elem] = set()
- for n in range(mesh.nodal_adjacency.neighbors_starts[elem], mesh.nodal_adjacency.neighbors_starts[elem + 1]):
+ starts = mesh.nodal_adjacency.neighbors_starts
+ for n in range(starts[elem], starts[elem + 1]):
adjacency_list[elem].add(mesh.nodal_adjacency.neighbors[n])
from pymetis import part_graph
--
GitLab