Create test_figures.py