Fixed bug in set_node_variable_number

Author: calleman21

exodus_helper/core.py

@@ -1819,6 +1819,9 @@ def set_node_variable_number(self, num_vars):
             dimensions = ('time_step', 'num_nod_var', 'num_nodes')
             self.dataset.createVariable(
                 'vals_nod_var', np.dtype('float'), dimensions=dimensions)
+            dimensions = ('num_nod_var', 'len_name')
+            self.dataset.createVariable(
+                'name_nod_var', np.dtype('S1'), dimensions=dimensions)
         return True
 
     def set_side_set_variable_number(self, num_vars):

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = 'poetry.core.masonry.api'
 
 [tool.poetry]
 name = 'exodus_helper'
-version = '1.5.1'
+version = '1.5.2'
 description = 'A package for manipulating ExodusII databases'
 license = 'BSD-3-Clause'
 authors = ['Coleman Alleman <callema@sandia.gov>']

tests/test_exodus_helper.py

@@ -1107,9 +1107,16 @@ def test_set_node_set_variable_truth_table():
     assert False
 
 
-@pytest.mark.unwritten
-def test_set_node_variable_number():
-    assert False
+def test_set_node_variable_number(dir_test_file):
+    file_path = os.path.join(dir_test_file, 'test_set_node_variable_number.g')
+    try:
+        mesh = exodus_helper.RectangularPrism(file_path, mode='w')
+        mesh.set_node_variable_number(1)
+        mesh.put_node_variable_name('test', 1)
+        assert mesh.dataset.dimensions['num_nod_var'].size == 1
+        assert mesh.get_node_variable_names()[0] == 'test'
+    finally:
+        os.remove(file_path)
 
 
 @pytest.mark.unwritten