From 738232a30e7c3a7837cbf9d2c5a352a3e3affa3e Mon Sep 17 00:00:00 2001
From: Zach Bateman <zkbateman@gmail.com>
Date: Tue, 2 Jun 2020 00:14:29 -0500
Subject: [PATCH] Add tests for map and for future shared-memory map

---
 tests/map_sharedmem_test.py | 35 +++++++++++++++++++++++++++++++++++
 tests/map_test.py           | 24 ++++++++++++++++++++++++
 2 files changed, 59 insertions(+)
 create mode 100644 tests/map_sharedmem_test.py
 create mode 100644 tests/map_test.py

diff --git a/tests/map_sharedmem_test.py b/tests/map_sharedmem_test.py
new file mode 100644
index 0000000..8525439
--- /dev/null
+++ b/tests/map_sharedmem_test.py
@@ -0,0 +1,35 @@
+import easy_multip
+import time
+import random
+import sys
+sys.path.insert(1, '..')
+
+args = [(1,2), (3,5), (6,2), (7,0), (7,1), (8,5), (5,2), (8,8), (4,7), (1,6), (1,1), (2,2), (6,6), (7,4), (7,3), (5,5)]
+
+def test_func(arg_tup):
+    x, y, ref_data = arg_tup
+    time.sleep(random.random() * 2)
+    return x + y + ref_data[5]
+    
+def test_func2(arg_tup, shared_mem):
+    x, y = arg_tup
+    time.sleep(random.random() * 2)
+    return x + y + shared_mem[5]
+
+
+if __name__ == '__main__':
+    REF_DATA = [5 for _ in range(10 ** 8)]
+
+    t0 = time.time()
+    data1 = [test_func((*arg, REF_DATA)) for arg in args]
+    print(f'Normal loop time: {round(time.time() - t0, 2)}')
+
+    t0 = time.time()
+    data2 = easy_multip.map(test_func, [(*arg, REF_DATA) for arg in args])
+    print(f'easy_multip.doloop time: {round(time.time() - t0, 2)}')
+    
+    t0 = time.time()
+    data3 = easy_multip.map(test_func2, [(arg) for arg in args], shared_memory_iterable=REF_DATA)
+    print(f'easy_multip.doloop time: {round(time.time() - t0, 2)}')
+    
+    print(f'data1 == data2 == data3 check: {data1 == data2 == data3}')
diff --git a/tests/map_test.py b/tests/map_test.py
new file mode 100644
index 0000000..ec3a5cb
--- /dev/null
+++ b/tests/map_test.py
@@ -0,0 +1,24 @@
+import easy_multip
+import time
+import random
+import sys
+sys.path.insert(1, '..')
+
+args = [(1,2), (3,5), (6,2), (7,0), (7,1), (8,5), (5,2), (8,8), (4,7), (1,6), (1,1), (2,2), (6,6), (7,4), (7,3), (5,5)]
+
+def test_func(arg_tup):
+    x, y = arg_tup
+    time.sleep(random.random() * 2)
+    return x + y
+
+
+if __name__ == '__main__':
+    t0 = time.time()
+    data1 = [test_func(arg) for arg in args]
+    print(f'Normal loop time: {round(time.time() - t0, 2)}')
+
+    t0 = time.time()
+    data2 = easy_multip.map(test_func, args)
+    print(f'easy_multip.doloop time: {round(time.time() - t0, 2)}')
+    
+    print(f'data1 == data2 check: {data1 == data2}')