Some tests

Adding some tests to the dataset model