* training pytests * add json sample file * add pytest compatibility * add compatibility with package test * add reference for pytorch version