From 4ea2e8c444ad151296878b6ad2b1a0dfc7b823c4 Mon Sep 17 00:00:00 2001
From: Holger Nahrstaedt <holger@nahrstaedt.de>
Date: Thu, 8 Mar 2018 13:47:16 +0100
Subject: [PATCH] fix wallet unit test

Limit coverage to python 3.6
---
 .circleci/config.yml |  2 +-
 beem/wallet.py       |  2 +-
 tox.ini              | 11 ++++++++++-
 3 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/.circleci/config.yml b/.circleci/config.yml
index 1459b3ee..87ced900 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -47,7 +47,7 @@ jobs:
           name: run tests
           command: |
             ./cc-test-reporter before-build
-            tox -e py36,pylint
+            tox -e py36
             ./cc-test-reporter after-build --exit-code $?
 
       - deploy:
diff --git a/beem/wallet.py b/beem/wallet.py
index 12dc1aeb..053cb8c5 100644
--- a/beem/wallet.py
+++ b/beem/wallet.py
@@ -262,7 +262,7 @@ class Wallet(object):
             # Try to decode as wif
             PrivateKey(encwif, prefix=self.prefix)
             return encwif
-        except ValueError:
+        except (ValueError, AssertionError):
             pass
         if self.locked():
             raise AssertionError()
diff --git a/tox.ini b/tox.ini
index 26a58c70..cfe93926 100644
--- a/tox.ini
+++ b/tox.ini
@@ -1,8 +1,17 @@
 [tox]
-envlist = py{27,34,35,36,37},lint,pyflakes
+envlist = py{27,34,35,36,37}
 skip_missing_interpreters = true
 
 [testenv]
+deps =
+    mock>=2.0.0
+    pytest
+    pytest-mock
+    parameterized
+commands =
+    pytest
+
+[testenv:py36]
 deps =
     mock>=2.0.0
     pytest
-- 
GitLab