Write (unit) tests

Problem/Motivation

It would be helpful for the module's stability to have tests. I think unit tests would be the right choice?

Steps to reproduce

Proposed resolution

Remaining tasks

User interface changes

API changes

Data model changes