Added tests for dataset handlers