* add tests * fixing tests * change names of it * resolve comments Co-authored-by: Dmitry Shibanov <v-dmshib@microsoft.com>