Clean up pytest code.

[crkrenn]

You can add an argument to assertTrue which will be printed if the test fails, this can be useful when using a helper function to test multiple things.

Originally posted by @daub1 in #6 (comment)

If I have time, I will look into pytest.parametrize. https://docs.pytest.org/en/stable/parametrize.html

Reply…