Improve code style

Author: gjbex

source-code/testing/Unittest/Newick/node_utils.py

@@ -1,18 +1,17 @@
 #!/usr/bin/env python
 '''module containing various functions for working with trees and nodes'''
 
-from node_parser import NodeParser
 import unittest
+from node_parser import NodeParser
 
 
 def depth(node):
     '''compute the depth of the given tree'''
     if node is None:
         return 0
-    elif node.is_leaf():
+    if node.is_leaf():
         return 1
-    else:
-        return 1 + max(list(map(depth, node.children())))
+    return 1 + max(list(map(depth, node.children())))
 
 
 def depth_first_iterator(node):
@@ -131,5 +130,6 @@ def test_deep_tree(self):
         tree = parser.node()
         self.assertEqual(nr_leaf_nodes(tree), 1)
 
+
 if __name__ == '__main__':
     unittest.main()