Created a small test (using pytest)

[nickmalleson]

Not sure if this will work outside of my local environment yet. It doesn't affect any other parts of the code though so you could pull it in and we can work through it.