Added .type() method to Node class. (#25)

MPh-py · Apr 6, 2021 · 5792aa4 · 5792aa4
1 parent 4442161
commit 5792aa4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/mph/node.py b/mph/node.py
@@ -177,6 +177,10 @@ def tag(self):
         """Returns the node's tag."""
         return str(self.java.tag()) if self.exists() else None
 
+    def type(self):
+        """Returns the node's feature type."""
+        return str(self.java.getType())
+
     def parent(self):
         """Returns the parent node."""
         if self.is_root():

diff --git a/tests/test_node.py b/tests/test_node.py
@@ -119,6 +119,12 @@ def test_tag():
     assert Node(model, 'functions/step').tag() == 'step1'
 
 
+def test_type():
+    assert Node(model, 'functions/step').type() == 'Step'
+    assert Node(model, 'functions/image').type() == 'Image'
+    assert Node(model, 'functions/table').type() == 'Interpolation'
+
+
 def test_parent():
     assert Node(model, 'functions/step').parent() == Node(model, 'functions')