From cd99d6b874d9d09b3bb87b8485cc787885af71f1 Mon Sep 17 00:00:00 2001
From: YurenHao0426 <blackhao0426@gmail.com>
Date: Tue, 13 Jan 2026 23:49:05 -0600
Subject: init commit

---
 files/tests/test_data_io.py | 11 +++++++++++
 1 file changed, 11 insertions(+)
 create mode 100644 files/tests/test_data_io.py

(limited to 'files/tests/test_data_io.py')

diff --git a/files/tests/test_data_io.py b/files/tests/test_data_io.py
new file mode 100644
index 0000000..1f2ccd8
--- /dev/null
+++ b/files/tests/test_data_io.py
@@ -0,0 +1,11 @@
+import torch
+from files.data_io.dataset_loader import get_dataloader
+
+
+def test_dataloader_shape():
+    """Smoke test: verify dataloader output shape."""
+    train_loader, _ = get_dataloader("data_io/configs/shd.yaml")
+    x, y = next(iter(train_loader))
+    assert isinstance(x, torch.Tensor)
+    assert x.ndim == 3
+    assert y.ndim == 1
-- 
cgit v1.2.3