Remove unused imports from tests.

Signed-off-by: Gary Pamparà <gpampara@gmail.com>