A pythonic `range` function for javascript

range.js

export function range(start, end) {
  if (end === undefined) {
    end = start
    start = 0
  }

  function *_range() {
    for(let i = start; i < end; i++) {
      yield i
    }
  }
  return [..._range()]
}

range.test.js

import assert from 'assert'
import { range } from './range'

describe('range', () => {
  it('works with 1 arg', () => {
    assert.deepEqual(
      range(2),
      [
        0, 1
      ]
    )
  })
  it('works with 2 args', () => {
    assert.deepEqual(
      range(2, 6),
      [
        2, 3, 4, 5,
      ]
    )
  })
})