Merge pull request #11 from RedFT/rounding_tests

Rounding tests
RedFT · Apr 7, 2019 · dad8dbe · dad8dbe
2 parents 14987fc + be003b5
commit dad8dbe
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/test/test_rounding.py b/test/test_rounding.py
@@ -0,0 +1,28 @@
+import numpy as np
+import hexy as hx
+
+def test_cube_round():
+    test_coords = np.array([
+        [1.1, -1.4, 0.3],
+        [3.3, 2.3, -5.4],
+        ]);
+
+    expected_coords = np.array([
+        [1, -1, 0],
+        [3, 2, -5],
+        ]);
+
+    assert(np.array_equal(hx.cube_round(test_coords), expected_coords))
+
+def test_axial_round():
+    test_coords = np.array([
+        [1.1, -1.4],
+        [3.3, 2.3],
+        ]);
+
+    expected_coords = np.array([
+        [1, -1],
+        [3, 2],
+        ]);
+
+    assert(np.array_equal(hx.axial_round(test_coords), expected_coords))