black: Fix python styling code findings

Python code should be aligned with black styling
format. This commit fixes any issues that have been
detected by black formatter.

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