--- a
+++ b/tests/test_dataset.py
@@ -0,0 +1,18 @@
+from unittest import TestCase
+
+from mediaug.dataset import Dataset
+from mediaug.download import get_data_cache
+
+
+class TestDataset(TestCase):
+    def test_load_sipakmed_cells(self):
+        ds = Dataset('/Users/sean/projects/cancerDetection/data/cells')
+        self.assertEqual(4049, ds.size)
+    
+    def test_load_sipakmed_slides(self):
+        ds = Dataset('/Users/sean/projects/cancerDetection/data/slides')
+        self.assertEqual(223, len(ds['dyskeratotic']))
+        self.assertEqual(966, ds.size)
+
+
+    
\ No newline at end of file