From 8687a5acbf5b09d85cc45f9ecd2a1df6079e295e Mon Sep 17 00:00:00 2001
From: Hari <harisankar995@gmail.com>
Date: Sun, 21 Jan 2024 11:30:09 +0100
Subject: [PATCH] Create test_heuristic.py

---
 test/test_heuristic.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 test/test_heuristic.py

diff --git a/test/test_heuristic.py b/test/test_heuristic.py
new file mode 100644
index 0000000..bf5f0f2
--- /dev/null
+++ b/test/test_heuristic.py
@@ -0,0 +1,34 @@
+""" Test heuristics """
+
+import math
+
+from pathfinding3d.core.heuristic import chebyshev, euclidean, manhattan, null, octile
+
+EPS = 1e-6
+
+
+def test_null():
+    assert null(1, -2, 3) == 0.0
+
+
+def test_manhattan():
+    assert manhattan(1, -2, 3) == 2.0
+
+
+def test_euclidean():
+    assert math.isclose(euclidean(1, -2, 3), math.sqrt(14), rel_tol=EPS)
+
+
+def test_chebyshev():
+    assert chebyshev(1, -2, 3) == 3.0
+
+
+def test_octile():
+    assert math.isclose(octile(1, 1, 1), math.sqrt(3), rel_tol=EPS)
+
+
+def test_octile2():
+    dx, dy, dz = 1, 2, 3
+    dmax, dmid, dmin = 3, 2, 1
+    expected_result = dmax + ((2**0.5 - 1) * dmid) + ((3**0.5 - 2**0.5) * dmin)
+    assert math.isclose(octile(dx, dy, dz), expected_result, rel_tol=EPS)