* Update python API for Memory * Run unit tests on CPU only * add missed import * Update tests * fix py tests