+added sample shape for tests

*fix device context