Unit tests

Test cases

Implementation with Pytest