new version of Pytorch dataset tested with EMNIST and KMNIST

Author: TLESORT

tests/test_dataset_attributes.py

@@ -1,6 +1,8 @@
 import pytest
 
 from continuum import datasets as cont_datasets
+from torchvision.datasets import EMNIST, KMNIST
+from continuum.datasets import PyTorchDataset
 
 ATTRS = ["get_data", "_download"]
 
@@ -11,3 +13,13 @@ def test_has_attr(dataset_name):
 
     for attr in ATTRS:
         assert hasattr(d, attr), (dataset_name, attr)
+
+
+@pytest.mark.slow
+def test_PytorchDataset_EMNIST(tmpdir):
+    dataset_train = PyTorchDataset(tmpdir, dataset_type=EMNIST, train=True, download=True, split='letters')
+
+
+@pytest.mark.slow
+def test_PytorchDataset_KMNIST(tmpdir):
+    dataset_train = PyTorchDataset(tmpdir, dataset_type=KMNIST, train=True, download=True)