1 files changed, 5 insertions, 1 deletions
diff --git a/tests/test_sftp.py b/tests/test_sftp.py
index f8fab1ce..6417ac90 100755
--- a/tests/test_sftp.py
+++ b/tests/test_sftp.py
@@ -25,13 +25,15 @@ do test file operations in (so no existing files will be harmed).
 
 from binascii import hexlify
 import os
+import sys
 import warnings
 import threading
 import unittest
 from tempfile import mkstemp
 
 import paramiko
-from paramiko.common import *
+from paramiko.py3compat import PY2, b, u, StringIO
+from paramiko.common import o777, o600, o666, o644
 from tests.stub_sftp import StubServer, StubSFTPServer
 from tests.loop import LoopSocket
 from tests.util import test_path
@@ -553,6 +555,7 @@ class SFTPTest (unittest.TestCase):
         with open(localname, 'wb') as f:
             f.write(text)
         saved_progress = []
+
         def progress_callback(x, y):
             saved_progress.append((x, y))
         sftp.put(localname, FOLDER + '/bunny.txt', progress_callback)
@@ -662,6 +665,7 @@ class SFTPTest (unittest.TestCase):
         with open(localname, 'w') as f:
             f.write(text)
         saved_progress = []
+
         def progress_callback(x, y):
             saved_progress.append((x, y))
         res = sftp.put(localname, FOLDER + '/bunny.txt', progress_callback, False)