Simplify the test_exp_graph script.