Accept Node instances in Model.export(). (#25)

MPh-py · Apr 6, 2021 · aea55f4 · aea55f4
1 parent 7894048
commit aea55f4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/mph/model.py b/mph/model.py
@@ -606,8 +606,6 @@ def export(self, node, file=None):
                 node = self/node
             else:
                 node = self/'exports'/node
-        else:
-            node = Node(node)
         if not node.exists():
             logger.warning('Node does not exist in model tree')
             return

diff --git a/tests/test_model.py b/tests/test_model.py
@@ -384,6 +384,15 @@ def test_export():
     model.export('field')
     assert (here/'field.txt').exists()
     (here/'field.txt').unlink()
+    assert not (here/'field.txt').exists()
+    model.export('exports/field')
+    assert (here/'field.txt').exists()
+    (here/'field.txt').unlink()
+    assert not (here/'field.txt').exists()
+    model.export(model/'exports'/'field')
+    assert (here/'field.txt').exists()
+    (here/'field.txt').unlink()
+    assert not (here/'field.txt').exists()
     assert not (here/'field2.txt').exists()
     model.export('exports/field', here/'field2.txt')
     assert (here/'field2.txt').exists()