Use tempfiles to create artificial hdf5 file of toy data for test

Author: c-salomonsen

tests/test_dataloaders.py

@@ -3,13 +3,18 @@
 
 def test_uspsdataset0_6():
     from pathlib import Path
+    from tempfile import TemporaryFile
 
+    import h5py
     import numpy as np
 
-    datapath = Path("data/USPS")
+    with TemporaryFile() as tf:
+        with h5py.File(tf, "w") as f:
+            f["train/data"] = np.random.rand(10, 16 * 16)
+            f["train/target"] = np.array([6, 5, 4, 3, 2, 1, 0, 0, 0, 0])
 
-    dataset = USPSDataset0_6(data_path=datapath, train=True)
-    assert len(dataset) == 5460
-    data, target = dataset[0]
-    assert data.shape == (1, 16, 16)
-    assert all(target == np.array([0, 0, 0, 0, 0, 0, 1]))
+        dataset = USPSDataset0_6(data_path=tf, train=True)
+        assert len(dataset) == 10
+        data, target = dataset[0]
+        assert data.shape == (1, 16, 16)
+        assert all(target == np.array([0, 0, 0, 0, 0, 0, 1]))