From 8303c58aa41248ffef99175e2707c6e63fe63591 Mon Sep 17 00:00:00 2001
From: protolambda <proto@protolambda.com>
Date: Tue, 21 May 2019 22:53:46 +0200
Subject: [PATCH] bugfix, make BLS wrapper propagate test output properly

---
 test_libs/pyspec/eth2spec/test/context.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/test_libs/pyspec/eth2spec/test/context.py b/test_libs/pyspec/eth2spec/test/context.py
index 7a8e959d0..2136a6053 100644
--- a/test_libs/pyspec/eth2spec/test/context.py
+++ b/test_libs/pyspec/eth2spec/test/context.py
@@ -76,6 +76,7 @@ def bls_switch(fn):
     def entry(*args, **kw):
         old_state = bls.bls_active
         bls.bls_active = kw.pop('bls_active', DEFAULT_BLS_ACTIVE)
-        fn(*args, **kw)
+        out = fn(*args, **kw)
         bls.bls_active = old_state
+        return out
     return entry