From 6b9195000eb5404c247288b384d7ca2bacc1ab23 Mon Sep 17 00:00:00 2001
From: Joris
Date: Sat, 6 Jun 2020 10:09:27 +0200
Subject: Add unit tests

---
 src/util/test_array.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 src/util/test_array.py

(limited to 'src/util/test_array.py')

diff --git a/src/util/test_array.py b/src/util/test_array.py
new file mode 100644
index 0000000..38759b9
--- /dev/null
+++ b/src/util/test_array.py
@@ -0,0 +1,12 @@
+from array import insert_position
+
+def test_insert_position():
+    assert insert_position(0, [], False) == 0
+    assert insert_position(1, [1, 2, 3], False) == 0
+    assert insert_position(2, [1, 2, 3], False) == 1
+    assert insert_position(3, [1, 2, 3], False) == 2
+    assert insert_position(8, [1, 2, 3], False) == 3
+    assert insert_position(8, [3, 2, 1], True) == 0
+    assert insert_position(3, [3, 2, 1], True) == 0
+    assert insert_position(2, [3, 2, 1], True) == 1
+    assert insert_position(1, [3, 2, 1], True) == 2
-- 
cgit v1.2.3