tests: context: Make legacy test to use ztest API

Use ztest APIs in test case to make output unified with other test
cases.

Signed-off-by: Punit Vara <punit.vara@intel.com>
3 files changed