Here goes tests, using Pytest for unitary testing is a common practice. You should also add tests for data and model
