Implement BMI v2.0 methods

Added get_input_item_count, get_output_item_count, and
get_grid_face_edges. Included tests.

Author: mdpiper

heat/bmi_heat.py

@@ -263,6 +263,14 @@ def get_component_name(self):
         """Name of the component."""
         return self._name
 
+    def get_input_item_count(self):
+        """Get names of input variables."""
+        return len(self._input_var_names)
+
+    def get_output_item_count(self):
+        """Get names of output variables."""
+        return len(self._output_var_names)
+
     def get_input_var_names(self):
         """Get names of input variables."""
         return self._input_var_names
@@ -323,6 +331,9 @@ def get_grid_node_count(self, grid):
     def get_grid_nodes_per_face(self, grid, nodes_per_face):
         raise NotImplementedError("get_grid_nodes_per_face")
 
+    def get_grid_face_edges(self, grid, face_edges):
+        raise NotImplementedError("get_grid_face_edges")
+
     def get_grid_x(self, grid, x):
         raise NotImplementedError("get_grid_x")
 

tests/test_grid_info.py

@@ -18,6 +18,17 @@ def test_grid_var_names():
     assert names == ("plate_surface__temperature",)
 
 
+def test_grid_var_item_count():
+    model = BmiHeat()
+    model.initialize()
+
+    count = model.get_input_item_count()
+    assert count == 1
+
+    count = model.get_output_item_count()
+    assert count == 1
+
+
 def test_grid_var_units():
     model = BmiHeat()
     model.initialize()