Pytest support

Did all the effort to accomplish #10