Updated test for YAMLDataSource