mathjs/test/unit-tests/function/arithmetic/nthRoots.test.js

// test nthRoots
import assert from 'assert'

import math from '../../../../src/bundleAny'
const complex = math.complex
const nthRoots = math.nthRoots

describe('nthRoots', function () {
  it('should return an array of Complex roots', function () {
    const roots = nthRoots(complex('-1'), 6)
    const roots1 = [
      complex({ r: 1, phi: Math.PI / 6 }),
      complex(0, 1),
      complex({ r: 1, phi: (5 * Math.PI) / 6 }),
      complex({ r: 1, phi: (7 * Math.PI) / 6 }),
      complex(0, -1),
      complex({ r: 1, phi: (11 * Math.PI) / 6 })
    ]

    roots.forEach(function (value, index, array) {
      assert.deepStrictEqual(value, roots1[index])
    })
  })

  it('should return the correct answer for Complex values', function () {
    const roots = nthRoots(complex(3, 4), 2)

    assert.deepStrictEqual(roots, [
      complex(2, 1),
      complex(-2.0000000000000004, -0.9999999999999999)
    ])
  })

  const twos = [
    complex(2, 0),
    complex(0, 2),
    complex(-2, 0),
    complex(0, -2)
  ]
  it('should return pure roots without artifacts', function () {
    const roots = nthRoots(complex('16'), 4)

    roots.forEach(function (value, index, array) {
      assert.deepStrictEqual(value, twos[index])
    })
  })

  it('should return roots for numeric arguments', function () {
    const roots = nthRoots(16, 4)

    roots.forEach(function (value, index, array) {
      assert.deepStrictEqual(value, twos[index])
    })
  })

  it('should return roots for string arguments', function () {
    const roots = nthRoots('16', 4)

    roots.forEach(function (value, index, array) {
      assert.deepStrictEqual(value, twos[index])
    })
  })

  it('should return zero exactly once', function () {
    const roots2 = nthRoots(0)
    const roots4 = nthRoots(0, 4)
    const roots8 = nthRoots(0, 8)
    assert.deepStrictEqual(roots2, [complex(0)])
    assert.deepStrictEqual(roots4, [complex(0)])
    assert.deepStrictEqual(roots8, [complex(0)])
  })
})