Add yaml support for test data