run tests through autopep8

Author: knipknap

demos/multiple/multiple.py

@@ -1,8 +1,9 @@
 #!/usr/bin/env python
-from Exscript            import Host
-from Exscript.util.file  import get_hosts_from_file, get_accounts_from_file
+from Exscript import Host
+from Exscript.util.file import get_hosts_from_file, get_accounts_from_file
 from Exscript.util.start import start
 
+
 def one(job, host, conn):
     # You can add a safehold based on the guess_os() method.
     if conn.guess_os() != 'ios':
@@ -17,6 +18,7 @@ def one(job, host, conn):
     conn.execute('show ip int brie')
     print "myvariable is", conn.get_host().get('myvariable')
 
+
 def two(job, host, conn):
     conn.autoinit()
     conn.execute('show interface POS1/0')
@@ -33,4 +35,4 @@ def two(job, host, conn):
 # The "max_threads" keyword indicates the maximum number of concurrent
 # connections.
 hosts = get_hosts_from_file('hostlist.txt')
-start(accounts, hosts, two, max_threads = 2)
+start(accounts, hosts, two, max_threads=2)

demos/quickstart/quickstart.py

@@ -1,7 +1,8 @@
 #!/usr/bin/env python
-from Exscript.util.match    import any_match
+from Exscript.util.match import any_match
 from Exscript.util.template import eval_file
-from Exscript.util.start    import quickstart
+from Exscript.util.start import quickstart
+
 
 def do_something(job, host, conn):
     conn.execute('ls -1')

demos/report/report.py

@@ -5,7 +5,8 @@
 from Exscript.util.file import get_hosts_from_file, get_accounts_from_file
 from Exscript.util.report import status, summarize
 
-logger = Logger() # Logs everything to memory.
+logger = Logger()  # Logs everything to memory.
+
 
 @log_to(logger)
 @autologin()

demos/template/template.py

@@ -1,10 +1,11 @@
 #!/usr/bin/env python
 from Exscript.util.template import eval_file
-from Exscript.util.start    import quickstart
+from Exscript.util.start import quickstart
+
 
 def do_something(job, host, conn):
     assert conn.guess_os() == 'shell'
     conn.execute('ls -1')
-    eval_file(conn, 'template.exscript', foobar = 'hello-world')
+    eval_file(conn, 'template.exscript', foobar='hello-world')
 
 quickstart('ssh://xpc3', do_something)

tests/Exscript/AccountManagerTest.py

@@ -1,16 +1,21 @@
-import sys, unittest, re, os.path, warnings
+import sys
+import unittest
+import re
+import os.path
+import warnings
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
 
 from Exscript.Account import Account
 from Exscript.AccountPool import AccountPool
 from Exscript.AccountManager import AccountManager
 
+
 class AccountManagerTest(unittest.TestCase):
     CORRELATE = AccountManager
 
     def setUp(self):
-        self.am      = AccountManager()
-        self.data    = {}
+        self.am = AccountManager()
+        self.data = {}
         self.account = Account('user', 'test')
 
     def testConstructor(self):
@@ -52,7 +57,7 @@ def match_cb(host):
 
     def testGetAccountFromHash(self):
         pool1 = AccountPool()
-        acc1  = Account('user1')
+        acc1 = Account('user1')
         pool1.add_account(acc1)
         self.am.add_pool(pool1)
 
@@ -123,7 +128,8 @@ def testReleaseAccounts(self):
         self.assert_(account1 in pool.unlocked_accounts)
         self.assert_(account2 in self.am.default_pool.unlocked_accounts)
 
+
 def suite():
     return unittest.TestLoader().loadTestsFromTestCase(AccountManagerTest)
 if __name__ == '__main__':
-    unittest.TextTestRunner(verbosity = 2).run(suite())
+    unittest.TextTestRunner(verbosity=2).run(suite())

tests/Exscript/AccountPoolTest.py

@@ -1,21 +1,25 @@
-import sys, unittest, re, os.path
+import sys
+import unittest
+import re
+import os.path
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
 
-from Exscript             import Account
+from Exscript import Account
 from Exscript.AccountPool import AccountPool
-from Exscript.util.file   import get_accounts_from_file
+from Exscript.util.file import get_accounts_from_file
+
 
 class AccountPoolTest(unittest.TestCase):
     CORRELATE = AccountPool
 
     def setUp(self):
-        self.user1     = 'testuser1'
+        self.user1 = 'testuser1'
         self.password1 = 'test1'
-        self.account1  = Account(self.user1, self.password1)
-        self.user2     = 'testuser2'
+        self.account1 = Account(self.user1, self.password1)
+        self.user2 = 'testuser2'
         self.password2 = 'test2'
-        self.account2  = Account(self.user2, self.password2)
-        self.accm      = AccountPool()
+        self.account2 = Account(self.user2, self.password2)
+        self.accm = AccountPool()
 
     def testConstructor(self):
         accm = AccountPool()
@@ -69,7 +73,7 @@ def testAcquireAccount(self):
         self.assert_(self.accm.n_accounts() == 5)
 
         for i in range(0, 2000):
-            # Each time an account is acquired a different one should be 
+            # Each time an account is acquired a different one should be
             # returned.
             acquired = {}
             for n in range(0, 5):
@@ -110,7 +114,8 @@ def testReleaseAccounts(self):
         self.assert_(account1 in pool.unlocked_accounts)
         self.assert_(account2 in pool.unlocked_accounts)
 
+
 def suite():
     return unittest.TestLoader().loadTestsFromTestCase(AccountPoolTest)
 if __name__ == '__main__':
-    unittest.TextTestRunner(verbosity = 2).run(suite())
+    unittest.TextTestRunner(verbosity=2).run(suite())

tests/Exscript/AccountTest.py

@@ -1,24 +1,28 @@
-import sys, unittest, re, os.path
+import sys
+import unittest
+import re
+import os.path
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
 
 from Exscript import Account, PrivateKey
 
+
 class AccountTest(unittest.TestCase):
     CORRELATE = Account
 
     def setUp(self):
-        self.user      = 'testuser'
+        self.user = 'testuser'
         self.password1 = 'test1'
         self.password2 = 'test2'
-        self.key       = PrivateKey()
-        self.account   = Account(self.user,
-                                 self.password1,
-                                 self.password2,
-                                 self.key)
+        self.key = PrivateKey()
+        self.account = Account(self.user,
+                               self.password1,
+                               self.password2,
+                               self.key)
 
     def testConstructor(self):
-        key     = PrivateKey()
-        account = Account(self.user, self.password1, key = key)
+        key = PrivateKey()
+        account = Account(self.user, self.password1, key=key)
         self.assertEqual(account.get_key(), key)
         self.assertEqual(account.get_password(),
                          account.get_authorization_password())
@@ -84,7 +88,8 @@ def testGetAuthorizationPassword(self):
     def testGetKey(self):
         self.assertEqual(self.key, self.account.get_key())
 
+
 def suite():
     return unittest.TestLoader().loadTestsFromTestCase(AccountTest)
 if __name__ == '__main__':
-    unittest.TextTestRunner(verbosity = 2).run(suite())
+    unittest.TextTestRunner(verbosity=2).run(suite())

tests/Exscript/FileLoggerTest.py

@@ -1,4 +1,7 @@
-import sys, unittest, re, os.path
+import sys
+import unittest
+import re
+import os.path
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
 
 from tempfile import mkdtemp
@@ -7,17 +10,19 @@
 from Exscript.FileLogger import FileLogger
 from LoggerTest import LoggerTest, FakeJob
 
+
 class FakeError(Exception):
     pass
 
+
 class FileLoggerTest(LoggerTest):
     CORRELATE = FileLogger
 
     def setUp(self):
         self.tempdir = mkdtemp()
-        self.logdir  = os.path.join(self.tempdir, 'non-existent')
-        self.logger  = FileLogger(self.logdir, clearmem = False)
-        self.job     = FakeJob('fake')
+        self.logdir = os.path.join(self.tempdir, 'non-existent')
+        self.logger = FileLogger(self.logdir, clearmem=False)
+        self.job = FakeJob('fake')
         self.logfile = os.path.join(self.logdir, 'fake.log')
         self.errfile = self.logfile + '.error'
 
@@ -32,7 +37,8 @@ def testConstructor(self):
 
     def testAddLog(self):
         log = LoggerTest.testAddLog(self)
-        self.assert_(os.path.isfile(self.logfile), 'No such file: ' + self.logfile)
+        self.assert_(
+            os.path.isfile(self.logfile), 'No such file: ' + self.logfile)
         self.failIf(os.path.exists(self.errfile))
         return log
 
@@ -82,7 +88,8 @@ def testLogSucceeded2(self):
         self.assert_(os.path.isfile(self.logfile))
         self.failIf(os.path.exists(self.errfile))
 
+
 def suite():
     return unittest.TestLoader().loadTestsFromTestCase(FileLoggerTest)
 if __name__ == '__main__':
-    unittest.TextTestRunner(verbosity = 2).run(suite())
+    unittest.TextTestRunner(verbosity=2).run(suite())

tests/Exscript/HostTest.py

@@ -1,26 +1,30 @@
-import sys, unittest, re, os.path
+import sys
+import unittest
+import re
+import os.path
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
 
-from Exscript          import Host, Account
+from Exscript import Host, Account
 from Exscript.util.url import Url
-from util.urlTest      import urls
+from util.urlTest import urls
+
 
 class HostTest(unittest.TestCase):
     CORRELATE = Host
 
     def setUp(self):
         self.host = Host('localhost')
-        self.host.set_all(dict(testarg = 1))
+        self.host.set_all(dict(testarg=1))
 
     def testConstructor(self):
         host = Host('localhost')
         self.assertEqual(host.get_protocol(), 'telnet')
-        host = Host('localhost', default_protocol = 'foo')
+        host = Host('localhost', default_protocol='foo')
         self.assertEqual(host.get_protocol(), 'foo')
 
         for url, result in urls:
             host = Host(url)
-            uri  = Url.from_string(url)
+            uri = Url.from_string(url)
             self.assertEqual(host.get_name(),    uri.hostname)
             self.assertEqual(host.get_address(), uri.hostname)
             self.assertEqual(host.get_uri(), str(uri))
@@ -35,7 +39,7 @@ def testSetUri(self):
     def testGetUri(self):
         for url, result in urls:
             host = Host(url)
-            uri  = Url.from_string(url)
+            uri = Url.from_string(url)
             self.assertEqual(host.get_uri(), str(uri))
 
     def testGetDict(self):
@@ -73,30 +77,31 @@ def testSetName(self):
         self.assertEqual(self.host.get_name(), 'testhost')
 
     def testGetName(self):
-        pass # Tested in testSetName().
+        pass  # Tested in testSetName().
 
     def testSetProtocol(self):
         self.assertEqual(self.host.get_protocol(), 'telnet')
         self.host.set_protocol('dummy')
         self.assertEqual(self.host.get_protocol(), 'dummy')
 
     def testGetProtocol(self):
-        pass # Tested in testSetProtocol().
+        pass  # Tested in testSetProtocol().
 
     def testSetOption(self):
         self.assertRaises(TypeError, self.host.set_option, 'test', True)
         self.assertEqual(self.host.get_options(), {})
         self.assertEqual(self.host.get_option('verify_fingerprint'), None)
-        self.assertEqual(self.host.get_option('verify_fingerprint', False), False)
+        self.assertEqual(
+            self.host.get_option('verify_fingerprint', False), False)
         self.host.set_option('verify_fingerprint', True)
         self.assertEqual(self.host.get_option('verify_fingerprint'), True)
         self.assertEqual(self.host.get_options(), {'verify_fingerprint': True})
 
     def testGetOption(self):
-        pass # Tested in testSetOption().
+        pass  # Tested in testSetOption().
 
     def testGetOptions(self):
-        pass # Tested in testSetOption().
+        pass  # Tested in testSetOption().
 
     def testSetTcpPort(self):
         self.assertEqual(self.host.get_tcp_port(), 23)
@@ -106,7 +111,7 @@ def testSetTcpPort(self):
         self.assertEqual(self.host.get_tcp_port(), 123)
 
     def testGetTcpPort(self):
-        pass # Tested in testSetTcpPort().
+        pass  # Tested in testSetTcpPort().
 
     def testSetAccount(self):
         account = Account('test')
@@ -115,7 +120,7 @@ def testSetAccount(self):
         self.assertEqual(self.host.get_account(), account)
 
     def testGetAccount(self):
-        pass # Tested in testSetAccount().
+        pass  # Tested in testSetAccount().
 
     def testSet(self):
         self.assertEqual(self.host.get('test'), None)
@@ -165,7 +170,8 @@ def testGet(self):
         self.assertEqual(self.host.get('test',  1), 3)
         self.assertEqual(self.host.get('test2', 1), 1)
 
+
 def suite():
     return unittest.TestLoader().loadTestsFromTestCase(HostTest)
 if __name__ == '__main__':
-    unittest.TextTestRunner(verbosity = 2).run(suite())
+    unittest.TextTestRunner(verbosity=2).run(suite())

tests/Exscript/LogTest.py

@@ -1,10 +1,14 @@
-import sys, unittest, re, os.path
+import sys
+import unittest
+import re
+import os.path
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
 
-from Exscript.Log    import Log
-from Exscript        import Host
+from Exscript.Log import Log
+from Exscript import Host
 from util.reportTest import FakeError
 
+
 class LogTest(unittest.TestCase):
     CORRELATE = Log
 
@@ -68,7 +72,8 @@ def testHasEnded(self):
         self.testSucceeded()
         self.assert_(self.log.has_ended())
 
+
 def suite():
     return unittest.TestLoader().loadTestsFromTestCase(LogTest)
 if __name__ == '__main__':
-    unittest.TextTestRunner(verbosity = 2).run(suite())
+    unittest.TextTestRunner(verbosity=2).run(suite())