Added basic unit tests of non-integrated functions

This commit is contained in:
Joel Collins 2020-11-25 15:25:17 +00:00
parent f54684b460
commit 5137d1b9dc
7 changed files with 228 additions and 46 deletions

View file

@ -33,7 +33,7 @@ def construct_grid(initial, step_sizes, n_steps, style="raster"):
if style == "spiral":
# deal with the centre image immediately
coord = initial
arr.append(initial)
arr.append([initial])
# for spiral, n_steps is the number of shells, and so only requires n_steps[0]
for i in range(2, n_steps[0] + 1):
arr.append([])
@ -66,16 +66,6 @@ def construct_grid(initial, step_sizes, n_steps, style="raster"):
return arr
def flatten_grid(grid):
"""
Convert a 3D list of scan positions into a flat list
of sequential positions.
"""
grid = list(itertools.chain(*grid))
return grid
### Capturing