From 1a16b1b6b9ae8a87775f0288aaad0cfda7b7af88 Mon Sep 17 00:00:00 2001
From: vdemedes <vdemedes@gmail.com>
Date: Sun, 20 Sep 2015 17:44:33 +0300
Subject: [PATCH 1/2] use co() for running generator functions

---
 lib/test.js  | 3 ++-
 package.json | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/lib/test.js b/lib/test.js
index 0883f57..b38e0c7 100644
--- a/lib/test.js
+++ b/lib/test.js
@@ -4,6 +4,7 @@ var Promise = require('bluebird');
 var setImmediate = require('set-immediate-shim');
 var fnName = require('fn-name');
 var assert = require('./assert');
+var co = require('co');
 
 function Test(title, fn) {
 	if (!(this instanceof Test)) {
@@ -16,7 +17,7 @@ function Test(title, fn) {
 	}
 
 	this.title = title || fnName(fn) || '[anonymous]';
-	this.fn = isGenerator(fn) ? Promise.coroutine(fn) : fn;
+	this.fn = isGenerator(fn) ? co.wrap(fn) : fn;
 	this.assertCount = 0;
 	this.planCount = null;
 	this.duration = null;
diff --git a/package.json b/package.json
index 247fd8e..d4b6ced 100644
--- a/package.json
+++ b/package.json
@@ -63,6 +63,7 @@
     "babel-core": "^5.8.23",
     "bluebird": "^2.9.34",
     "chalk": "^1.0.0",
+    "co": "^4.6.0",
     "figures": "^1.3.5",
     "fn-name": "^2.0.0",
     "globby": "^3.0.1",

From a0de27799cc7e64481531f620940b6ff9d83f0c3 Mon Sep 17 00:00:00 2001
From: vdemedes <vdemedes@gmail.com>
Date: Sun, 20 Sep 2015 17:45:17 +0300
Subject: [PATCH 2/2] blacklist regenerator to not transpile generator
 functions

---
 lib/babel.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/lib/babel.js b/lib/babel.js
index b0a3ccc..e675e04 100644
--- a/lib/babel.js
+++ b/lib/babel.js
@@ -3,7 +3,8 @@
 var resolveFrom = require('resolve-from');
 
 var options = {
-	only: /(test|test\-.+|test\/.+)\.js$/
+	only: /(test|test\-.+|test\/.+)\.js$/,
+	blacklist: ['regenerator']
 };
 
 try {