Getting ready for unit testing with pytest