Add tests for chart and data file generation