wip: add dataset unit test