From 5699738b9b1d140dbd39ed77b038461fbd9f436c Mon Sep 17 00:00:00 2001
From: Oleksii Trekhleb <trehleb@gmail.com>
Date: Mon, 16 Apr 2018 22:18:05 +0300
Subject: [PATCH] Add primality tests.

---
 .../primality-tests/__test__/primalityTest.js | 24 -----------------
 .../__test__/trialDivision.test.js            | 26 ++++++++++++++++++-
 2 files changed, 25 insertions(+), 25 deletions(-)
 delete mode 100644 src/algorithms/math/primality-tests/__test__/primalityTest.js

diff --git a/src/algorithms/math/primality-tests/__test__/primalityTest.js b/src/algorithms/math/primality-tests/__test__/primalityTest.js
deleted file mode 100644
index 72a456be..00000000
--- a/src/algorithms/math/primality-tests/__test__/primalityTest.js
+++ /dev/null
@@ -1,24 +0,0 @@
-/**
- * @param {function(n: number)} testFunction
- */
-export default function primalityTest(testFunction) {
-  expect(testFunction(1)).toBeTruthy();
-  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();
-}
diff --git a/src/algorithms/math/primality-tests/__test__/trialDivision.test.js b/src/algorithms/math/primality-tests/__test__/trialDivision.test.js
index c9c8ea1e..9e5b67ec 100644
--- a/src/algorithms/math/primality-tests/__test__/trialDivision.test.js
+++ b/src/algorithms/math/primality-tests/__test__/trialDivision.test.js
@@ -1,5 +1,29 @@
 import trialDivision from '../trialDivision';
-import primalityTest from './primalityTest';
+
+/**
+ * @param {function(n: number)} testFunction
+ */
+function primalityTest(testFunction) {
+  expect(testFunction(1)).toBeTruthy();
+  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', () => {