Introduce testing framework using pytest

Just added it!