浏览代码

use manual python calls instead of pytest

justheuristic 4 年之前
父节点
当前提交
e606a27a53
共有 1 个文件被更改,包括 2 次插入0 次删除
  1. 2 0
      tests/test_averaging.py

+ 2 - 0
tests/test_averaging.py

@@ -1,4 +1,5 @@
 import asyncio
 import asyncio
+import contextlib
 import gc
 import gc
 import random
 import random
 import time
 import time
@@ -22,6 +23,7 @@ from test_utils.dht_swarms import launch_dht_instances
 logger = get_logger(__name__)
 logger = get_logger(__name__)
 
 
 
 
+@contextlib.contextmanager
 def cleanup_children():
 def cleanup_children():
     yield
     yield