reformat code, simplify tests (needless args)

Author: funbringer

testgres/connection.py

@@ -17,7 +17,6 @@
     default_dbname as _default_dbname, \
     default_username as _default_username
 
-
 # export these exceptions
 InternalError = pglib.InternalError
 ProgrammingError = pglib.ProgrammingError
@@ -36,11 +35,7 @@ class NodeConnection(object):
     Transaction wrapper returned by Node
     """
 
-    def __init__(self,
-                 node,
-                 dbname=None,
-                 username=None,
-                 password=None):
+    def __init__(self, node, dbname=None, username=None, password=None):
 
         # Use default user if not specified
         dbname = dbname or _default_dbname()

testgres/node.py

@@ -69,11 +69,7 @@ def __bool__(self):
 
 
 class PostgresNode(object):
-    def __init__(self,
-                 name=None,
-                 port=None,
-                 base_dir=None,
-                 use_logging=False):
+    def __init__(self, name=None, port=None, base_dir=None, use_logging=False):
         """
         Create a new node manually.
 
@@ -609,7 +605,6 @@ def cleanup(self, max_attempts=3):
 
     @method_decorator(positional_args_hack(['query'], ['dbname', 'query']))
     def psql(self,
-             *,
              query=None,
              filename=None,
              dbname=None,
@@ -668,7 +663,6 @@ def psql(self,
     @method_decorator(positional_args_hack(['dbname', 'query']))
     def safe_psql(self,
                   query,
-                  *,
                   dbname=None,
                   username=None,
                   input=None):
@@ -747,7 +741,6 @@ def restore(self, filename, dbname=None, username=None):
     @method_decorator(positional_args_hack(['dbname', 'query']))
     def poll_query_until(self,
                          query,
-                         *,
                          dbname=None,
                          username=None,
                          max_attempts=0,
@@ -816,7 +809,6 @@ def poll_query_until(self,
     @method_decorator(positional_args_hack(['dbname', 'query']))
     def execute(self,
                 query,
-                *,
                 dbname=None,
                 username=None,
                 password=None,

testgres/utils.py

@@ -250,7 +250,7 @@ def positional_args_hack(*special_cases):
     'special_cases' into named args.
 
     Example:
-        @positional_args_hack([['dbname', 'query']])
+        @positional_args_hack(['abc'], ['def', 'abc'])
         def some_api_func(...)
 
     This is useful for compatibility.