59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
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")
|