Introduce testing framework using pytest