javascript-algorithms-and-data-structures/src/algorithms/math/primality-test/__test__/trialDivision.test.js

Algorithms and data-structures implemented on JavaScript

import trialDivision from '../trialDivision';

/**
 * @param {function(n: number)} testFunction
 */
function primalityTest(testFunction) {
  expect(testFunction(1)).toBeFalsy();
  expect(testFunction(2)).toBeTruthy();
  expect(testFunction(3)).toBeTruthy();
  expect(testFunction(5)).toBeTruthy();
  expect(testFunction(11)).toBeTruthy();
  expect(testFunction(191)).toBeTruthy();
  expect(testFunction(191)).toBeTruthy();
  expect(testFunction(199)).toBeTruthy();

  expect(testFunction(-1)).toBeFalsy();
  expect(testFunction(0)).toBeFalsy();
  expect(testFunction(4)).toBeFalsy();
  expect(testFunction(6)).toBeFalsy();
  expect(testFunction(12)).toBeFalsy();
  expect(testFunction(14)).toBeFalsy();
  expect(testFunction(25)).toBeFalsy();
  expect(testFunction(192)).toBeFalsy();
  expect(testFunction(200)).toBeFalsy();
  expect(testFunction(400)).toBeFalsy();
}

describe('trialDivision', () => {
  it('should detect prime numbers', () => {
    primalityTest(trialDivision);
  });
});