Given that this is for testing purposes I feel it would be better to place this into a `test_.py` file as opposed to production code file.