Add more test cases for YamlDataSet