cleaned and limited output validation in test_figures
removed remaining spaces in docstrings and superfluous imports