Fix new stuff from tests to work with TreeNodes

psh/commands.py

@@ -1,6 +1,6 @@
 from io import StringIO
 
-from tree import TreeNode
+from psh.tree import TreeNode
 
 
 class BaseCommand(object):

psh/example_cmd.py

@@ -1,6 +1,6 @@
 from psh.commands import BaseCommand, register_cmd
 
-from tree import TreeNode
+from psh.tree import TreeNode
 
 
 @register_cmd("example")

psh/raw_commands.py

@@ -2,8 +2,7 @@ import shlex
 
 from psh.formatters import Printer
 from psh.commands import BaseCommand
-
-from tree import TreeNode
+from psh.tree import TreeNode
 
 
 class RawCommand(BaseCommand):

test/utils.py

@@ -14,7 +14,7 @@ class TestFormatter(BaseCommand):
     def call(self):
         input_generator = self.get_input_generator()
         for line in input_generator:
-            self.buffer.write(line.decode('utf-8'))
+            self.buffer.write(line.data.decode('utf-8'))
         return None
 
     def get_data(self):