[bug 838000] fix unit test to use dataset.