Improve test utility

sindresorhus · Mar 10, 2024 · aa78dbf · aa78dbf
1 parent 7ccab70
commit aa78dbf
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/test/helpers/index.js b/test/helpers/index.js
@@ -1,12 +1,8 @@
-import {Duplex} from 'node:stream';
+import {Duplex, Readable} from 'node:stream';
 
-export const createStream = streamDef => {
-	const generator = typeof streamDef === 'function' ? streamDef : function * () {
-		yield * streamDef;
-	};
-
-	return Duplex.from(generator);
-};
+export const createStream = streamDef => typeof streamDef === 'function'
+	? Duplex.from(streamDef)
+	: Readable.from(streamDef);
 
 // Tests related to big buffers/strings can be slow. We run them serially and
 // with a higher timeout to ensure they do not randomly fail.