Unit testing with the pytest module