Added basic unit tests of non-integrated functions

2020-11-25 15:25:17 +00:00 · 2020-11-25 15:25:17 +00:00 · 5137d1b9dc
commit 5137d1b9dc
parent f54684b460
7 changed files with 228 additions and 46 deletions
--- a/tests/test_extension_scan.py
+++ b/tests/test_extension_scan.py
@ -0,0 +1,59 @@
+from pprint import pprint
+
+from openflexure_microscope.api.default_extensions import scan
+
+
+def test_construct_grid_raster():
+    grid = scan.construct_grid((0, 0), (100, 100), (3, 3), style="raster")
+    assert grid == [
+        [(0, 0), (0, 100), (0, 200)],
+        [(100, 0), (100, 100), (100, 200)],
+        [(200, 0), (200, 100), (200, 200)],
+    ]
+
+
+def test_construct_grid_snake():
+    grid = scan.construct_grid((0, 0), (100, 100), (3, 3), style="snake")
+    assert grid == [
+        [(0, 0), (0, 100), (0, 200)],
+        [(100, 200), (100, 100), (100, 0)],
+        [(200, 0), (200, 100), (200, 200)],
+    ]
+
+
+def test_construct_grid_spiral():
+    grid = scan.construct_grid((0, 0), (100, 100), (3, 0), style="spiral")
+    assert grid == [
+        [(0, 0)],
+        [
+            (0, 100),
+            (100, 100),
+            (100, 0),
+            (100, -100),
+            (0, -100),
+            (-100, -100),
+            (-100, 0),
+            (-100, 100),
+        ],
+        [
+            (-100, 200),
+            (0, 200),
+            (100, 200),
+            (200, 200),
+            (200, 100),
+            (200, 0),
+            (200, -100),
+            (200, -200),
+            (100, -200),
+            (0, -200),
+            (-100, -200),
+            (-200, -200),
+            (-200, -100),
+            (-200, 0),
+            (-200, 100),
+            (-200, 200),
+        ],
+    ]
+
+    # Ensure second n_steps value makes no difference
+    assert grid == scan.construct_grid((0, 0), (100, 100), (3, 3), style="spiral")