Pytest for package for data processing