Fixed bug in get_idxs_node

Author: calleman21

exodus_helper/core.py

@@ -1269,8 +1269,8 @@ def get_idxs_node(self, ids_node):
         Returns:
             A `numpy.ndarray` of node indices (int)
         """
-        elem_id_map = self.get_elem_id_map()
-        return np.array([np.where(elem_id_map == i)[0][0] for i in ids_node])
+        node_id_map = self.get_node_id_map()
+        return np.array([np.where(node_id_map == i)[0][0] for i in ids_node])
 
     def get_info_records(self) -> list:
         """Get a list of info records where each entry in the list is one info

tests/test_exodus_helper.py

@@ -654,6 +654,24 @@ def test_get_idxs_elem_start_blk(mesh):
     assert idxs[0] == 0 and idxs[1] == 2
 
 
+def test_get_idxs_elem(mesh):
+    ids = mesh.get_elem_id_map()
+    np.random.shuffle(ids)
+    mesh.put_elem_id_map(ids)
+    idxs = mesh.get_idxs_elem(ids)
+    assert np.all(idxs == np.arange(mesh.get_num_elems()))
+    mesh.put_elem_id_map(np.sort(ids))
+
+
+def test_get_idxs_node(mesh):
+    ids = mesh.get_node_id_map()
+    np.random.shuffle(ids)
+    mesh.put_node_id_map(ids)
+    idxs = mesh.get_idxs_node(ids)
+    assert np.all(idxs == np.arange(mesh.get_num_nodes()))
+    mesh.put_node_id_map(np.sort(ids))
+
+
 def test_get_info_records(mesh):
     assert 'info_records' in mesh.dataset.variables
     assert 'num_info' in mesh.dataset.dimensions