python-tests: Exit test from one point

It is always better to have a single return/exit
from a function/test in which it would be easier for
users to understand.

Signed-off-by: Ahmed Ismail <Ahmed.Ismail@arm.com>