Tests [TODO]

# Overview

Write unit tests for each function

# Completion criteria

- [ ] For each function, add a "todo" issue to create a test for the function, describing (a) what the function is supposed to do and (b) how it should be tested. If there are any edge cases or tricky scenarios to watch out for, those should be noted too

# Resources (optional)

- [pytest](https://docs.pytest.org/en/7.3.x/): framework for writing and running tests
- [GitHub actions](https://github.com/features/actions): can be used to automate tests when pull requests are merged
- [davos](https://github.com/ContextLab/davos): archive contains useful code for running tests that involve notebooks

# Discussion points

Adding issues doesn't require coding, but it'd be good to have at least one "coding" person work on this task to help define useful tests.