feat(range): support negative steps

innocenzi · Oct 12, 2023 · ed52261 · ed52261
1 parent b1c7ff8
commit ed52261
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/array.test.ts b/src/array.test.ts
@@ -30,6 +30,7 @@ it('range', () => {
 	expect(range(2)).toEqual([0, 1])
 	expect(range(2, 5)).toEqual([2, 3, 4])
 	expect(range(2, 10, 2)).toEqual([2, 4, 6, 8])
+	expect(range(3, 0, -1)).toEqual([3, 2, 1])
 })
 
 it('partition', () => {

diff --git a/src/array.ts b/src/array.ts
@@ -154,7 +154,7 @@ export function range(...args: any): number[] {
 
 	const arr: number[] = []
 	let current = start
-	while (current < stop) {
+	while (step > 0 ? (current < stop) : (current > stop)) {
 		arr.push(current)
 		current += step || 1
 	}