[FEATURE REQUEST] Initialise unit tests

**Is your feature request related to a problem? Please describe.**
Difficult to test new versions of code or refactoring

**Describe the solution you'd like**
Pytest has ways to test bits of code with code coverage

**Describe alternatives you've considered**
There may be other options but Pytest seems to be the most used

**Additional context**
NA