[IMP]:Improve test case and check context is not None