Add a test case that uses datasets

[khatchad]

Add test case for https://github.com/wala/ML/issues/89. We lose track of the tensors in the dataset.