Add tests for utility functions

[meganrm]

Testing display components can be a little contrived, but I would for sure suggest adding unit tests for utility functions like this one.

Originally posted by @tyler-foster in #196 (comment)