Adding tests for merge_datasets