Add testing architecture with py.test