Add py.test framework