fix test import err

Revolution1 · Mar 3, 2018 · d01aca9 · d01aca9
1 parent f0d5c43
commit d01aca9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tests/etcd_go_cli.py b/tests/etcd_go_cli.py
@@ -62,15 +62,17 @@ def etcdctl(*args, **kwargs):
     cmd.extend(args)
     p = Popen(cmd, stdout=PIPE, env=envs)
     out, err = p.communicate()
-    if not p.returncode == 0:
+    if p.returncode != 0:
         raise RuntimeError(err)
     return out
 
+
+NO_ETCD_SERVICE = True
 try:
     if ETCDCTL_PATH and etcdctl('--dial-timeout=0.2s endpoint health'):
         NO_ETCD_SERVICE = False
 except:
-    NO_ETCD_SERVICE = True
+    pass
 
 if __name__ == '__main__':
     print(etcdctl('get foo'))
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -1,5 +1,4 @@
 import pytest
-import six
 
 from etcd3.client import Etcd3APIClient
 from etcd3.errors import Etcd3APIError