public code v1

test/data_reader/test_data_reader_short.py

@@ -0,0 +1,16 @@
+import pytest
+
+from pygrex.config import cfg
+from pygrex.data_reader.data_reader import DataReader
+
+
+@pytest.fixture
+def data_reader():
+    return DataReader(**cfg.data.testdata)
+
+
+def test_import(data_reader):
+    assert data_reader.num_user == 249
+    assert data_reader.num_item == 551
+    assert data_reader.dataset.shape[0] == 1000
+    assert data_reader.dataset.shape[1] == 4