Integrated pytest

First attempt to use pytest