Use pytests for testing